This commit is contained in:
EmptyCrown 2023-02-05 17:57:03 -08:00
parent 07e97723da
commit 07f1ab0acd
4 changed files with 5 additions and 8 deletions

View File

@ -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"

View File

@ -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

View File

@ -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):

View File

@ -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 = []