Spaces:

cloudyuga
/

Invoice-Extractor-Gradio-App

Sleeping

pratikshahp commited on Jan 21

Commit

2d0804c

verified ·

1 Parent(s): 55726a8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ def get_gemini_response(input_prompt, uploaded_file_path, query):
         image_parts = [{"mime_type": mime_type, "data": image_data}]
         # Load the Gemini model and get the response
-        model = genai.GenerativeModel("gemini-pro-vision")
         response = model.generate_content([input_prompt, image_parts[0], query])
         return response.text
     except Exception as e:

         image_parts = [{"mime_type": mime_type, "data": image_data}]
         # Load the Gemini model and get the response
+        model = genai.GenerativeModel("gemini-1.5-flash")
         response = model.generate_content([input_prompt, image_parts[0], query])
         return response.text
     except Exception as e: