Spaces:
Runtime error
Runtime error
import gradio as gr | |
from transformers import pipeline | |
# Use a pipeline as a high-level helper | |
model = pipeline("text-generation", model="cognitivecomputations/samantha-mistral-7b") | |
def predict(text): | |
return model(text) | |
demo = gr.Interface(fn=predict, inputs="text", outputs="label") | |
demo.launch() |