Spaces:
Sleeping
Sleeping
import gradio as gr | |
from transformers import pipeline | |
pipe = pipeline("text-generation", model="42dot/42dot_LLM-PLM-1.3B") | |
def chat_with_model(prompt): | |
response = pipe(prompt) | |
return response[0]['generated_text'] | |
interface = gr.Interface(fn=chat_with_model, inputs="text", outputs="text") | |
interface.launch() |