File size: 326 Bytes
3ce9b95
7e7f4e3
3ce9b95
7e7f4e3
 
3ce9b95
7e7f4e3
 
3ce9b95
7e7f4e3
 
1
2
3
4
5
6
7
8
9
10
11
12
import gradio as gr
from transformers import pipeline

# Load your large model
model = pipeline("text-generation", model="cognitivecomputations/dolphin-2.9-llama3-8b")

def generate_text(prompt):
    return model(prompt)[0]['generated_text']

demo = gr.Interface(fn=generate_text, inputs="text", outputs="text")
demo.launch()