Update app.py

app.py

@@ -12,8 +12,11 @@ def chat(question):
 
     inputs = tokenizer(prompt, return_tensors="pt")
     output = model.generate(inputs["input_ids"], max_new_tokens=100)
-
-    print(tokenizer.decode(output[0].tolist(), skip_special_tokens=True))
+    
+    response = tokenizer.decode(output[0].tolist(), skip_special_tokens=True)
+    print(response)
+    
+    return response
 
 iface = gr.Interface(fn=chat,
                      inputs=gr.inputs.Textbox(label="Enter your text"),
@@ -21,4 +24,4 @@ iface = gr.Interface(fn=chat,
                      title="Chat with Surrey County Council InfoBot")
 
 # index = construct_index("docs")
-iface.launch()
+iface.launch()