Spaces:
Running
Running
Krish Patel
commited on
Commit
·
0e5ed12
1
Parent(s):
25bb7ca
Debugging1
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def generate_knowledge_graph_viz(text, nlp, tokenizer, model):
|
|
42 |
# Update knowledge graph
|
43 |
kg_builder.update_knowledge_graph(text, not is_fake, nlp)
|
44 |
|
45 |
-
# Randomly select subset of edges (e.g.
|
46 |
edges = list(kg_builder.knowledge_graph.edges())
|
47 |
selected_edges = random.sample(edges, k=int(len(edges) * 0.1))
|
48 |
|
@@ -154,10 +154,14 @@ def main():
|
|
154 |
if gemini_result and gemini_result.get('gemini_analysis'):
|
155 |
break
|
156 |
|
157 |
-
except Exception:
|
158 |
-
|
|
|
159 |
|
160 |
retry_count += 1
|
|
|
|
|
|
|
161 |
|
162 |
# Use default values if all retries failed
|
163 |
if not gemini_result:
|
|
|
42 |
# Update knowledge graph
|
43 |
kg_builder.update_knowledge_graph(text, not is_fake, nlp)
|
44 |
|
45 |
+
# Randomly select subset of edges (e.g. 10% of edges)
|
46 |
edges = list(kg_builder.knowledge_graph.edges())
|
47 |
selected_edges = random.sample(edges, k=int(len(edges) * 0.1))
|
48 |
|
|
|
154 |
if gemini_result and gemini_result.get('gemini_analysis'):
|
155 |
break
|
156 |
|
157 |
+
except Exception as e:
|
158 |
+
st.error(f"Gemini API error: {str(e)}")
|
159 |
+
print(f"Gemini error: {str(e)}")
|
160 |
|
161 |
retry_count += 1
|
162 |
+
import time
|
163 |
+
time.sleep(1) # Add a 1-second delay between retries
|
164 |
+
|
165 |
|
166 |
# Use default values if all retries failed
|
167 |
if not gemini_result:
|