Update main.py

main.py

@@ -39,9 +39,9 @@ def generate(prompt, history, temperature=0.2, max_new_tokens=30000, top_p=0.95,
     formatted_prompt = format_prompt(prompt, history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
 
-    # Accumula l'output in una lista
-    output_list = []
+    # Accumula l'output in una stringa anziché una lista
+    output_text = ""
     for response in stream:
-        output_list.append(response.token.text)
+        output_text += response.token.text
 
-    return iter(output_list)  # Restituisci la lista come un iteratore
+    return output_text  # Restituisci l'intero testo generato come una stringa