kurry commited on
Commit
faa2f16
·
1 Parent(s): 5db3809

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration
4
  def generate_text(input_text):
5
  # Load pre-trained model and tokenizer
6
  model_name = 'kurry/t5_small_finetuned'
7
- model = T5ForConditionalGeneration.from_pretrained(model_name, from_tf=True)
8
  tokenizer = T5Tokenizer.from_pretrained(model_name)
9
 
10
  # Generate summary
 
4
  def generate_text(input_text):
5
  # Load pre-trained model and tokenizer
6
  model_name = 'kurry/t5_small_finetuned'
7
+ model = T5ForConditionalGeneration.from_pretrained(model_name)
8
  tokenizer = T5Tokenizer.from_pretrained(model_name)
9
 
10
  # Generate summary