Update app.py

app.py

@@ -117,7 +117,7 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
                         opts.append(lines[i+z].split(" ",1)[1])
                 except Exception:
                     pass
-        if ": " in line:
+        if ": " in line[:12]:
             try:
                 lab_1 = line.split(": ")[0]
                 
@@ -147,12 +147,18 @@ def clear_fn():
     return None,None
 
 base_stats=[
-    {"Health":100,"Power":20,"Strength":24},
+    {},
 ]
 text_stats='''*******************
 Health: 100
 Power: 20
 Strength: 24
+Intelligence: 15
+Charisma: 15
+Dexterity: 18
+Wisdom: 15
+Compassion: 15
+Courage: 15
 *******************    
 '''