Spaces:

BigSalmon
/

Paraphrase

Runtime error

BigSalmon commited on Dec 4, 2021

Commit

b197998

1 Parent(s): 5012e22

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import torch
 from transformers import T5ForConditionalGeneration,T5Tokenizer, AutoTokenizer, AutoModelForSeq2SeqLM
 import streamlit as st
 st.title("Paraphrase")
-model_name = st.text_input("Pick a Model", "seduerr/t5-pawraphrase")
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained("t5-base")

 from transformers import T5ForConditionalGeneration,T5Tokenizer, AutoTokenizer, AutoModelForSeq2SeqLM
 import streamlit as st
 st.title("Paraphrase")
+model_name = st.text_input("Pick a Model", "geckos/pegasus-fined-tuned-on-paraphrase")
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained("t5-base")