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() |