Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -109,7 +109,7 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
|
|
109 |
lab_1 = line.split(": ")[0]
|
110 |
|
111 |
skill_1 = line.split(": ")[1].split(" ")[0]
|
112 |
-
skill_1
|
113 |
skill_1=int(skill_1)
|
114 |
skill_dict[lab_1]=skill_1
|
115 |
#skill ={lab_1:skill_1}
|
|
|
109 |
lab_1 = line.split(": ")[0]
|
110 |
|
111 |
skill_1 = line.split(": ")[1].split(" ")[0]
|
112 |
+
skill_1.strip("</s>")
|
113 |
skill_1=int(skill_1)
|
114 |
skill_dict[lab_1]=skill_1
|
115 |
#skill ={lab_1:skill_1}
|