Update app.py

app.py

@@ -110,9 +110,9 @@ def generate(prompt, history,max_new_tokens,health,seed,temperature=temperature,
         if ":" in line:
             for z in range(1,5):
                 try:
-                    if f'{z}' in lines[i+z+1]:
-                        print(lines[i+z+1])
-                        opts.append(lines[i+z+1])
+                    if f'{z}.' in lines[i+z]:
+                        print(lines[i+z])
+                        opts.append(lines[i+z])
                 except Exception as e:
                     print(e)
                     pass