Update app.py

app.py

@@ -10,8 +10,7 @@ def generate_question(description):
     prompt = f"Only generate a factual and relevant question about this memory: {description}"
     inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True)
     outputs = model.generate(**inputs, max_new_tokens=64)
-    question = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return question
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 interface = gr.Interface(
     fn=generate_question,
@@ -19,4 +18,5 @@ interface = gr.Interface(
     outputs=gr.Textbox(label="Generated Question"),
 )
 
-app = gr.mount_gradio_app(app=None, blocks=interface, path="/")
+demo = gr.Interface(fn=generate_question, inputs="text", outputs="text")
+demo.launch()