Spaces:
Runtime error
Runtime error
Simon Salmon
commited on
Commit
·
f29adcd
1
Parent(s):
8c5035b
Update app.py
Browse files
app.py
CHANGED
@@ -1,10 +1,9 @@
|
|
1 |
import torch
|
2 |
-
from transformers import T5ForConditionalGeneration,T5Tokenizer
|
3 |
import streamlit as st
|
4 |
|
5 |
-
|
6 |
-
model =
|
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 |
|