Spaces:

Go-Raw
/

Perspective

Sleeping

Go-Raw commited on Jul 3

Commit

4763d2b

verified ·

1 Parent(s): d6054ba

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import gradio as gr
 # Load T0pp (instruction-tuned)
-model_name = "Xenova/flan-t5-base"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)

 import gradio as gr
 # Load T0pp (instruction-tuned)
+model_name = "arunadiraju/my_awesome_qa_model"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)