Update app.py

app.py

@@ -113,7 +113,7 @@ 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]:
+                    if f'{z}' in lines[i+z][:6]:
                         print(lines[i+z])
                         opts.append(lines[i+z])
                 except Exception as e: