Update main.py
Browse files
main.py
CHANGED
|
@@ -93,10 +93,11 @@ async def get_data(input_file: UploadFile = File(...)):
|
|
| 93 |
text += page.extract_text()
|
| 94 |
|
| 95 |
if text=="":
|
|
|
|
| 96 |
text = vision(file_content)
|
| 97 |
else:
|
| 98 |
raise HTTPException(status_code=400, detail="Unsupported file type")
|
| 99 |
-
|
| 100 |
# Call Gemini (or another model) to extract required data
|
| 101 |
prompt = f"""This is CV data: {text.strip()}
|
| 102 |
I want only:
|
|
|
|
| 93 |
text += page.extract_text()
|
| 94 |
|
| 95 |
if text=="":
|
| 96 |
+
print("vision called")
|
| 97 |
text = vision(file_content)
|
| 98 |
else:
|
| 99 |
raise HTTPException(status_code=400, detail="Unsupported file type")
|
| 100 |
+
print(text,"=========================================")
|
| 101 |
# Call Gemini (or another model) to extract required data
|
| 102 |
prompt = f"""This is CV data: {text.strip()}
|
| 103 |
I want only:
|