Update main.py

main.py

@@ -33,4 +33,5 @@ async def generate_response(item: Validation):
     output = llm(prompt, max_tokens = item.max_tokens,temperature = item.temperature, echo=True)
     
     # Extract and return the text from the response
-    return output['choices'][0]['text']
+    return output['choices'][0]['text']
+