Update app.py

app.py

@@ -34,21 +34,18 @@ def generate_text(message, history):
     input_prompt += "[|Umano|] " + message + "\n[|Assistente|]"
 
     print(input_prompt)
-
-    karg = {
-    'input_prompt': input_prompt,
-    'temperature': 0.15,
-    'top_p': 0.1,
-    'top_k': 40, 
-    'repeat_penalty': 1.1,
-    'max_tokens': 1024,
-    'stop': [
+    
+    output = saiga(input_prompt,
+    temperature: 0.15,
+    top_p: 0.1,
+    top_k: 40, 
+    repeat_penalty: 1.1,
+    max_tokens: 1024,
+    stop: [
         "[|Umano|]",
         "[|Assistente|]",
     ],
-    'stream': True
-       }
-    output = saiga(**karg)
+    stream: True)
     
     for out in output:
         stream = copy.deepcopy(out)
@@ -59,7 +56,7 @@ def generate_text(message, history):
 
 def generate_text_Dante(message, history):
     temp = ""
-    input_prompt = "Conversazione tra umano ed un assistente AI di nome saiga-7b\n"
+    input_prompt = ""
     for interaction in history:
         input_prompt += "[|Umano|] " + interaction[0] + "\n"
         input_prompt += "[|Assistente|]" + interaction[1]
@@ -67,22 +64,18 @@ def generate_text_Dante(message, history):
     input_prompt += "[|Umano|] " + message + "\n[|Assistente|]"
 
     print(input_prompt)
-
-    karg = {
-    'input_prompt': input_prompt,
-    'temperature': 0.15,
-    'top_p': 0.1,
-    'top_k': 40, 
-    'repeat_penalty': 1.1,
-    'max_tokens': 1024,
-    'stop': [
+           
+    output = dante(input_prompt,
+    temperature: 0.15,
+    top_p: 0.1,
+    top_k: 40, 
+    repeat_penalty: 1.1,
+    max_tokens: 1024,
+    stop: [
         "[|Umano|]",
         "[|Assistente|]",
     ],
-    'stream': True
-       }
-           
-    output = dante(**karg)
+    stream: True)
     
     for out in output:
         stream = copy.deepcopy(out)