Update app.py

app.py

@@ -109,6 +109,7 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
                 lab_1 = line.split(": ")[0]
                 
                 skill_1 = line.split(": ")[1].split(" ")[0]
+                skill_1 = line.strip("<\s>")
                 skill_1=int(skill_1)
                 skill_dict[lab_1]=skill_1
                 #skill ={lab_1:skill_1}