Spaces:

HeheBoi0769
/

Nexus_NLP_model

Running

HeheBoi0769 commited on Apr 4

Commit

d4f888d

verified ·

1 Parent(s): d207b6f

Update final.py

Changed gemini model

Files changed (1) hide show

final.py CHANGED Viewed

@@ -70,7 +70,7 @@ class KnowledgeGraphBuilder:
 def setup_gemini():
     """Initialize Gemini model"""
     genai.configure(api_key=os.getenv("GEMINI_API"))
-    model = genai.GenerativeModel('models/gemini-1.5-flash')
     return model
 def predict_with_model(text, tokenizer, model):

 def setup_gemini():
     """Initialize Gemini model"""
     genai.configure(api_key=os.getenv("GEMINI_API"))
+    model = genai.GenerativeModel('models/gemini-2.0-flash')
     return model
 def predict_with_model(text, tokenizer, model):