updated

app.py

@@ -17,7 +17,8 @@ pipe = pipeline("text-generation", "meta-llama/Meta-Llama-3-8B-Instruct", torch_
 def generate(prompt):
     response = pipe(prompt, max_new_tokens=512)
     # r = response[0]['generated_text'][-1]['content']
-    return response
+    r = response[0]['generated_text']
+    return r
 
 gr.Interface(
     fn=generate,