Downloads · 30 days
0
ozzygraniol/MOVELIGHT
MOVELIGHT is a machine learning model from ozzygraniol. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
--- license: from flask import Flask, rendertemplate, request from transformers import pipeline
Downloads · 30 days
0
Access
Public
Updated Feb 19, 2024
Repo size
—
Likes
0
Public
Click a slice to open those files.
Other1.5 KB · 69%
From the Hugging Face model README
license: from flask import Flask, render_template, request from transformers import pipeline
app = Flask(name)
model = pipeline("text-generation", model="distilgpt2")
@app.route("/", methods=["GET", "POST"]) def index(): if request.method == "POST": # Get user input prompt = request.form["prompt"] # Generate text based on user input generated_text = model(prompt, max_length=50, do_sample=True)[0]['generated_text'] return render_template("index.html", prompt=prompt, generated_text=generated_text) return render_template("index.html")
if name == "main": app.run(debug=True)