Update main.py

main.py

@@ -42,9 +42,9 @@ def generate(item: Item):
         # Format the prompt
         formatted_prompt = format_prompt(f"{item.system_prompt}, {item.prompt}", item.history)
 
-        # Call text_generation on your model
+        # Call text_generation on your model (correct argument: formatted_prompt)
         stream = client.text_generation(
-            inputs=formatted_prompt,
+            formatted_prompt,  # Use the formatted prompt directly
             **generate_kwargs,
             stream=True,
         )