Simon Salmon commited on
Commit
f78dae5
·
1 Parent(s): 67647ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
 
 
1
  import streamlit as st
2
- from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
3
-
4
 
5
  model_name = "ramsrigouthamg/t5_paraphraser"
6
  model = T5ForConditionalGeneration.from_pretrained('ramsrigouthamg/t5_paraphraser')
 
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')