balaramas commited on
Commit
a5d0aa6
1 Parent(s): fca7f66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,10 +3,10 @@ import gradio as gr
3
  import subprocess
4
 
5
  subprocess.check_call(["pip", "install", "transformers"])
6
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
7
-
8
 
9
 
 
10
 
11
  tokenizer = AutoTokenizer.from_pretrained("balaramas/mbart-enhiriser")
12
  model = AutoModelForSeq2SeqLM.from_pretrained("balaramas/mbart-enhiriser")
 
3
  import subprocess
4
 
5
  subprocess.check_call(["pip", "install", "transformers"])
6
+ subprocess.check_call(["pip", "install", "torch"])
 
7
 
8
 
9
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
10
 
11
  tokenizer = AutoTokenizer.from_pretrained("balaramas/mbart-enhiriser")
12
  model = AutoModelForSeq2SeqLM.from_pretrained("balaramas/mbart-enhiriser")