leooj's picture
Update app.py
fd4c1b2
raw
history blame contribute delete
318 Bytes
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()