Upload main.py

main.py

@@ -32,7 +32,7 @@ def generate(text: str):
     llama2 q4 backend
     """
     output = llm(text)
-    return {"output": output[0]["generated_text"]}
+    return {"output": output}
 
 
 def prompt(question):