Update app.py

app.py

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