File size: 535 Bytes
8058203 ad74568 |
1 2 3 4 5 6 7 8 9 10 11 |
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() |