Jaane commited on
Commit
32f5239
·
verified ·
1 Parent(s): 2ebab6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ os.environ['TRANSFORMERS_NO_ADVISORY_WARNINGS'] = '1'
12
  def random_state(seed):
13
  torch.manual_seed(seed)
14
  if torch.cuda.is_available():
15
- torch.cuda.manual_seed_all(1234)
16
 
17
  # Initialize Parrot model
18
  parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")
 
12
  def random_state(seed):
13
  torch.manual_seed(seed)
14
  if torch.cuda.is_available():
15
+ torch.cuda.manual_seed_all(1234)
16
 
17
  # Initialize Parrot model
18
  parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")