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:
"""Init reader."""
from transformers import (
AutoTokenizer,
VisionEncoderDecoderModel,
ViTFeatureExtractor,
)
from transformers import (AutoTokenizer, VisionEncoderDecoderModel,
ViTFeatureExtractor)
model = VisionEncoderDecoderModel.from_pretrained(
"nlpconnect/vit-gpt2-image-captioning"

View File

@ -7,7 +7,6 @@ Currently cannot load documents.
from typing import Any, List, Optional
import requests
from gpt_index.readers.base import BaseReader
from gpt_index.readers.schema.base import Document
from gpt_index.response.schema import Response, SourceNode

View File

@ -1,11 +1,11 @@
"""Read Arxiv Papers."""
import os
import hashlib
import os
from typing import List, Optional
from gpt_index import download_loader
from gpt_index.readers.base import BaseReader
from gpt_index.readers.schema.base import Document
from gpt_index import download_loader
class ArxivReader(BaseReader):

View File

@ -26,6 +26,7 @@ class PubmedReader(BaseReader):
List[Document]: A list of Document objects.
"""
import xml.etree.ElementTree as xml
import requests
pubmed_search = []