Update app.py

app.py

@@ -1,14 +1,10 @@
-from flask import Flask, request, jsonify
-from transformers import pipeline
+import gradio as gr
 
-app = Flask(__name__)
-model = pipeline('text-generation', model='Erfan11/Neuracraft')
+def respond_to_input(user_input):
+    return f"You said: {user_input}"
 
-@app.route('/generate', methods=['POST'])
-def generate_text():
-    input_text = request.json.get('input_text', '')
-    result = model(input_text, max_length=100)
-    return jsonify(result)
+with gr.Blocks() as demo:
+    gr.Textbox(label="Input", placeholder="Type something...").submit(respond_to_input)
+    gr.Textbox(label="Output")
 
-if __name__ == '__main__':
-    app.run(debug=True, host='0.0.0.0', port=5000)
+demo.launch()