Omnibus commited on
Commit
8a1dea2
1 Parent(s): dfc43a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = line.strip("<\s>")
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}