Debate_v2 / app.py
anirudh-sub's picture
Added an import
a83fa2d
raw
history blame
495 Bytes
import gradio as gr
import transformers
from transformers import AutoTokenizer
import torch
model = "anirudh-sub/debate_model_v2.1"
tokenizer = AutoTokenizer.from_pretrained(model)
pipeline = transformers.pipeline(
"text-generation",
model=model,
torch_dtype=torch.float16,
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()