Update app.py

app.py

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