Downloads · 30 days
0
slimen/chat_slimen
chat_slimen is a machine learning model from slimen. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
from transformers import pipeline, setseed from flask import Flask, jsonify, request, rendertemplate
Downloads · 30 days
0
Access
Public
Updated May 7, 2023
Repo size
—
Likes
0
Public
Click a slice to open those files.
Other1.5 KB · 72%
From the Hugging Face model README
from transformers import pipeline, set_seed from flask import Flask, jsonify, request, render_template
app = Flask(name) generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
@app.route('/') def home(): return render_template('index.html')
@app.route('/get_answer', methods=['POST']) def get_answer(): set_seed(42) question = request.form['question'] answer = generator(question, max_length=50, do_sample=True, temperature=0.7)[0]['generated_text'] return jsonify({'answer': answer})
if name == 'main': app.run(debug=True)