Update app.py

app.py

@@ -104,6 +104,10 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     skill_dict={}
     new_stat="*******************\n"
     for i,line in enumerate(lines):
+        if "Choices:" in line:
+            for z in range(1,10):
+                if f'{z}' in lines[i+z]:
+                    print(lines[i+z].split(" ")[1])
         if ": " in line:
             try:
                 lab_1 = line.split(": ")[0]