Updated gemini model

app.py

@@ -44,7 +44,7 @@ def generate_knowledge_graph_viz(text, nlp, tokenizer, model):
 
     # Randomly select subset of edges (e.g. 10% of edges)
     edges = list(kg_builder.knowledge_graph.edges())
-    selected_edges = random.sample(edges, k=int(len(edges) * 0.1))
+    selected_edges = random.sample(edges, k=int(len(edges) * 0.3))
     
     # Create a new graph with selected edges
     selected_graph = nx.DiGraph()

final.py

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