Jay7478 commited on
Commit
cc695b7
·
1 Parent(s): b71da86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -90,7 +90,8 @@ import nltk
90
  import math
91
  import torch
92
 
93
- model_name = "afnanmmir/t5-base-abstract-to-plain-language-1"
 
94
  max_input_length = 1024
95
  max_output_length = 256
96
  min_output_length = 64
 
90
  import math
91
  import torch
92
 
93
+ # model_name = "afnanmmir/t5-base-abstract-to-plain-language-1"
94
+ model_name = "afnanmmir/fabiochiu/t5-base-medium-title-generation"
95
  max_input_length = 1024
96
  max_output_length = 256
97
  min_output_length = 64