Spaces:
Sleeping
Sleeping
Omar ID EL MOUMEN
commited on
Commit
·
d588a4d
1
Parent(s):
9a324f3
Fix attempt #1
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ class Query(BaseModel):
|
|
52 |
class DocumentID(BaseModel):
|
53 |
doc_id: str
|
54 |
|
55 |
-
class
|
56 |
url: str
|
57 |
|
58 |
@app.post("/search")
|
@@ -130,8 +130,8 @@ async def extract_random_arxiv_pdf(query: Query):
|
|
130 |
return await extract_arxiv_pdf(random.choice(list(pubs["message"].keys())))
|
131 |
|
132 |
@app.post("/extract_pdf/url")
|
133 |
-
async def extract_pdf(url:
|
134 |
-
pdf_req = requests.get(url
|
135 |
if pdf_req.status_code == 200:
|
136 |
pdf_data = BytesIO(pdf_req.content)
|
137 |
doc = fitz.open(stream=pdf_data, filetype="pdf")
|
|
|
52 |
class DocumentID(BaseModel):
|
53 |
doc_id: str
|
54 |
|
55 |
+
class WebPDF(BaseModel):
|
56 |
url: str
|
57 |
|
58 |
@app.post("/search")
|
|
|
130 |
return await extract_arxiv_pdf(random.choice(list(pubs["message"].keys())))
|
131 |
|
132 |
@app.post("/extract_pdf/url")
|
133 |
+
async def extract_pdf(url: WebPDF):
|
134 |
+
pdf_req = requests.get(url)
|
135 |
if pdf_req.status_code == 200:
|
136 |
pdf_data = BytesIO(pdf_req.content)
|
137 |
doc = fitz.open(stream=pdf_data, filetype="pdf")
|