Update app.py

app.py

@@ -67,9 +67,16 @@ def generate(
     lines = output.strip().strip("\n").split("\n")
     #history=""
     for i,line in enumerate(lines):
-        if line.startswith("choices:"):
-            for z in range(1,5):
-                print(lines[i+z])
+        if line.startswith("1. "):
+            print(line)
+        if line.startswith("2. "):
+            print(line)
+        if line.startswith("3. "):
+            print(line)
+        if line.startswith("4. "):
+            print(line)
+        if line.startswith("5. "):
+            print(line)            
     return output