Update app.py

app.py

@@ -109,7 +109,7 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     option_drop=[]
     new_stat="*******************\n"
     for i,line in enumerate(lines):
-        if ":" in line:
+        if "Choices:" in line:
             for z in range(1,5):
                 try:
                     if f'{z}' in lines[i+z]:
@@ -139,9 +139,9 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
 
     if history:
         history.append((prompt,output))
-        yield history,stats,skills,opts
+        yield history,stats,skills,option_drop
     else:
-        yield [(prompt,output)],stats,skills,opts
+        yield [(prompt,output)],stats,skills,option_drop
 
 def clear_fn():
     return None,None