Update app.py

app.py

@@ -101,6 +101,7 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
        
     lines = output.strip().strip("\n").split("\n")
     skills=[]
+    skill_dict={}
     new_stat="*******************"
     for i,line in enumerate(lines):
         if ": " in line:
@@ -109,16 +110,18 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
                 
                 skill_1 = line.split(": ")[1].split(" ")[0]
                 skill_1=int(skill_1)
-                skill ={lab_1:skill_1}
-                skills.append(skill)
+                skill_dict[lab_1]=skill_1
+                #skill ={lab_1:skill_1}
                 
-                new_stat += f'{lab_1}: {skill_1}'
+                new_stat += f'{lab_1}: {skill_1}\n'
                 
                 print(skills)
             except Exception as e:
                 print (f'--Error :: {e}')
             print(f'Line:: {line}')
-    stats=new_stat
+    skills.append(skill_dict)
+
+    stats=new_stat+="*******************\n"
     if history:
         history.append((prompt,output))
         yield "",history,stats,skills