test-t2t / app.py
pow27's picture
Update app.py
ad74568 verified
raw
history blame contribute delete
535 Bytes
import gradio as gr
def generate_story(prompt):
# Это функция, которая будет вызывать вашу модель
# Например, вы можете использовать transformers для вызова модели
from transformers import pipeline
generator = pipeline("text-generation", model="molbal/horrorllama3-8b-v1.0")
return generator(prompt, max_length=100)[0]["generated_text"]
interface = gr.Interface(fn=generate_story, inputs="text", outputs="text")
interface.launch()