Update app.py

app.py

@@ -109,8 +109,11 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     for i,line in enumerate(lines):
         if "Choices:" in line:
             for z in range(1,10):
-                if f'{z}' in lines[i+z]:
-                    print(lines[i+z].split(" ")[1])
+                try:
+                    if f'{z}' in lines[i+z]:
+                        print(lines[i+z].split(" ",1)[1])
+                except Exception:
+                    pass
         if ": " in line:
             try:
                 lab_1 = line.split(": ")[0]