Debate_v2 / app.py
anirudh-sub's picture
Trying to debug app.py
b45ed0e
raw
history blame
451 Bytes
import gradio as gr
import transformers
from transformers import AutoTokenizer
model = "anirudh-sub/debate_model_v2.1"
tokenizer = AutoTokenizer.from_pretrained(model)
pipeline = transformers.pipeline(
"text-generation",
model=model,
device_map="auto",
)
def greet(name):
return "Hello " + name + "!"
demo = gr.Interface(
fn=greet,
inputs=gr.Textbox(lines=2, placeholder="Name Here..."),
outputs="text",
)
demo.launch()