Spaces:
Runtime error
Runtime error
File size: 555 Bytes
87df5c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from flask import Flask, render_template, request
import gradio as gr
from transformers import pipeline
app = Flask(__name__)
# Load chatbot model
chatbot = pipeline("text2text-generation", model="facebook/blenderbot-2B")
@app.route('/')
def home():
return render_template('index.html')
@app.route('/chat')
def chat():
return render_template('chat.html')
@app.post('/generate_text')
def generate_text(input_text):
response = chatbot(input_text)
return response[0]['generated_text']
if __name__ == '__main__':
app.run(debug=True)
|