mirror of
https://github.com/run-llama/llama-hub.git
synced 2025-11-10 14:54:46 +00:00
Format
This commit is contained in:
parent
07e97723da
commit
07f1ab0acd
@ -17,11 +17,8 @@ class PptxReader(BaseReader):
|
|||||||
|
|
||||||
def __init__(self) -> None:
|
def __init__(self) -> None:
|
||||||
"""Init reader."""
|
"""Init reader."""
|
||||||
from transformers import (
|
from transformers import (AutoTokenizer, VisionEncoderDecoderModel,
|
||||||
AutoTokenizer,
|
ViTFeatureExtractor)
|
||||||
VisionEncoderDecoderModel,
|
|
||||||
ViTFeatureExtractor,
|
|
||||||
)
|
|
||||||
|
|
||||||
model = VisionEncoderDecoderModel.from_pretrained(
|
model = VisionEncoderDecoderModel.from_pretrained(
|
||||||
"nlpconnect/vit-gpt2-image-captioning"
|
"nlpconnect/vit-gpt2-image-captioning"
|
||||||
|
|||||||
@ -7,7 +7,6 @@ Currently cannot load documents.
|
|||||||
from typing import Any, List, Optional
|
from typing import Any, List, Optional
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
from gpt_index.readers.base import BaseReader
|
from gpt_index.readers.base import BaseReader
|
||||||
from gpt_index.readers.schema.base import Document
|
from gpt_index.readers.schema.base import Document
|
||||||
from gpt_index.response.schema import Response, SourceNode
|
from gpt_index.response.schema import Response, SourceNode
|
||||||
|
|||||||
@ -1,11 +1,11 @@
|
|||||||
"""Read Arxiv Papers."""
|
"""Read Arxiv Papers."""
|
||||||
import os
|
|
||||||
import hashlib
|
import hashlib
|
||||||
|
import os
|
||||||
from typing import List, Optional
|
from typing import List, Optional
|
||||||
|
|
||||||
|
from gpt_index import download_loader
|
||||||
from gpt_index.readers.base import BaseReader
|
from gpt_index.readers.base import BaseReader
|
||||||
from gpt_index.readers.schema.base import Document
|
from gpt_index.readers.schema.base import Document
|
||||||
from gpt_index import download_loader
|
|
||||||
|
|
||||||
|
|
||||||
class ArxivReader(BaseReader):
|
class ArxivReader(BaseReader):
|
||||||
|
|||||||
@ -26,6 +26,7 @@ class PubmedReader(BaseReader):
|
|||||||
List[Document]: A list of Document objects.
|
List[Document]: A list of Document objects.
|
||||||
"""
|
"""
|
||||||
import xml.etree.ElementTree as xml
|
import xml.etree.ElementTree as xml
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
pubmed_search = []
|
pubmed_search = []
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user