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