Simon Salmon commited on
Commit
f29adcd
·
1 Parent(s): 8c5035b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -1,10 +1,9 @@
1
  import torch
2
- from transformers import T5ForConditionalGeneration,T5Tokenizer
3
  import streamlit as st
4
 
5
- model_name = "ramsrigouthamg/t5_paraphraser"
6
- model = T5ForConditionalGeneration.from_pretrained('ramsrigouthamg/t5_paraphraser')
7
- tokenizer = T5Tokenizer.from_pretrained('ramsrigouthamg/t5_paraphraser')
8
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
9
  model = model.to(device)
10
 
 
1
  import torch
2
+ from transformers import T5ForConditionalGeneration,T5Tokenizer, AutoTokenizer, AutoModelForSeq2SeqLM
3
  import streamlit as st
4
 
5
+ tokenizer = AutoTokenizer.from_pretrained("Vamsi/T5_Paraphrase_Paws")
6
+ model = AutoModelForSeq2SeqLM.from_pretrained("Vamsi/T5_Paraphrase_Paws")
 
7
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
8
  model = model.to(device)
9