AkshayaKeerthi commited on
Commit
7b701ad
·
verified ·
1 Parent(s): db0b04f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -6
app.py CHANGED
@@ -1,7 +1,20 @@
1
- import tkinter as tk
 
 
2
 
3
- root = tk.Tk()
4
- root.title("Hello, Tkinter!")
5
- label = tk.Label(root, text="Hello, Tkinter!")
6
- label.pack()
7
- root.mainloop()
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # app.py
2
+ from flask import Flask, render_template, request, jsonify
3
+ from transformers import pipeline
4
 
5
+ app = Flask(__name__)
6
+ model = pipeline('conversational')
7
+
8
+ @app.route('/')
9
+ def home():
10
+ return render_template('index.html')
11
+
12
+ @app.route('/predict', methods=['POST'])
13
+ def predict():
14
+ user_input = request.json.get('message')
15
+ result = model(user_input)
16
+ response_text = result[0]['generated_text']
17
+ return jsonify({'response': response_text})
18
+
19
+ if __name__ == '__main__':
20
+ app.run(host='0.0.0.0', port=5000)