Hugging Face's logo

Spaces:
heerjtdev
/
answer_validator
Sleeping

App Files Community
heerjtdev commited on
Commit
b879237
·
verified ·
1 Parent(s): b2f46de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -15,7 +15,10 @@ if not GEMINI_API_KEY:
15
 
16
  genai.configure(api_key=GEMINI_API_KEY)
17
 
18
- MODEL = genai.GenerativeModel("gemini-pro")
 
 
 
19
 
20
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
21
  EMBED_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
 
15
 
16
  genai.configure(api_key=GEMINI_API_KEY)
17
 
18
+
19
+
20
+
21
+ MODEL = genai.GenerativeModel("gemini-2.0-flash")
22
 
23
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
24
  EMBED_MODEL = "sentence-transformers/all-MiniLM-L6-v2"