Update app.py

app.py

@@ -8,12 +8,12 @@ from transformers import AutoTokenizer, TFAutoModelForSeq2SeqLM, AutoModelForSeq
 import numpy as np
 
 loaded_model = AutoModelForSequenceClassification.from_pretrained("runaksh/financial_summary_T5_base")
-loaded_tokenizer = AutoTokenizer.from_pretrained("runaksh/financial_summary_T5_base")
+loaded_tokenizer = AutoTokenizer.from_pretrained("runaksh/financial_summary_T5_base",local_files_only=True)
 
 # Function for generating summary
 def generate_summary(text,min_length=55,max_length=80):
   text = "summarize: "+text
-  input = tokenizer(text,max_length=512,truncation=True,return_tensors='pt').input_ids
+  input = tokenizer(text,max_length=512,truncation=True,return_tensors='tf').input_ids
   op=model.generate(input,min_length=min_length,max_length=max_length)
   decoded_op = tokenizer.batch_decode(op,skip_special_tokens=True)
   return decoded_op