Update app.py

app.py

@@ -109,15 +109,15 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     option_drop=[]
     new_stat="*******************\n"
     for i,line in enumerate(lines):
-        if "Choices:" in line:
-            for z in range(1,5):
-                try:
-                    if f'{z}' in lines[i+z]:
-                        print(lines[i+z].split(" ",1)[1])
-                        opts.append(lines[i+z].split(" ",1)[1])
-                except Exception as e:
-                    print(e)
-                    pass
+        #if ":" in line:
+        #    for z in range(1,5):
+        #        try:
+        if f'{z}.' in line[:6]:
+            print(lines[i+z].split(" ",1)[1])
+            opts.append(lines[i+z].split(" ",1)[1])
+        #        except Exception as e:
+        #            print(e)
+        #            pass
         if ": " in line[:20]:
             try:
                 lab_1 = line.split(": ")[0]