Update app.py

app.py

@@ -71,7 +71,7 @@ def predict(temp, limit, text):
     prompt = text
     out = inferance(prompt, mergedModel, tokenizer, temp = 1.0, limit = 200)
     display = extract_answer(out)
-    return out
+    return display
 
 pred = gr.Interface(
     predict,