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