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