🌘w🌖

app.py

@@ -99,10 +99,17 @@ def talk(message, history):
     time.sleep(1)
     # Initialize an empty string to store the generated text
     partial_text = ""
-    for new_text in streamer:
-        if new_text is not None:
-            partial_text += new_text
-            yield partial_text
+    i = 0
+    while t.is_alive():
+        try:
+            for new_text in streamer:
+                if new_text is not None:
+                    partial_text += new_text
+                    yield partial_text
+        except Exception as e:
+            print(f"retry number {i}\ LOGS:\n")
+            i+=1
+            print(e, e.args)
     partial_text += resources
     yield partial_text