galactica2_6.7b / app.py
sushmitxo's picture
Create app.py
2043da0
raw
history blame contribute delete
No virus
506 Bytes
import gradio as gr
from transformers import pipeline
def text_generation_model(input_text):
model = pipeline("text-generation", model="facebook/galactica-6.7b")
output_text = model(input_text)
return output_text
input_text_interface = gr.inputs.Textbox(placeholder="Enter text here")
output_text_interface = gr.outputs.Textbox()
interface = gr.Interface(fn=text_generation_model, inputs=input_text_interface, outputs=output_text_interface, title="Text Generation Model")
interface.launch()