Update app.py

app.py

@@ -13,7 +13,7 @@ loaded_tokenizer = AutoTokenizer.from_pretrained("runaksh/financial_summary_T5_b
 # 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="tf").input_ids
+  input = tokenizer(text,max_length=512,truncation=True,return_tensors='pt').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
@@ -37,4 +37,4 @@ iface = gradio.Interface(fn=generate_summary,
                          description=description
                          )
 
-iface.launch(debug = True, from_tf=True)
+iface.launch(debug = True)