Spaces:
Runtime error
Runtime error
File size: 577 Bytes
7b701ad 0a7b476 7b701ad 5a18b7c 7b701ad |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# app.py
from flask import Flask, render_template, request, jsonify
from transformers import pipeline
app = Flask(__name__)
model = pipeline('conversational', model='microsoft/DialoGPT-small', revision='main')
@app.route('/')
def home():
return render_template('index.html')
@app.route('/predict', methods=['POST'])
def predict():
user_input = request.json.get('message')
result = model(user_input)
response_text = result[0]['generated_text']
return jsonify({'response': response_text})
if __name__ == '__main__':
app.run(host='0.0.0.0', port=5000)
|