AfnanAl commited on
Commit
536c27c
·
1 Parent(s): 5e6a4d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,9 +16,9 @@ import os
16
 
17
  tokenizer = T5Tokenizer.from_pretrained("AfnanAl/mT5small-ArabicSummary")
18
 
19
- model_name="AfnanAl/mT5small-ArabicSummary"
20
- custom_output_path="AfnanAl/summaryAonnx"
21
- modelO=get_onnx_model(model_name, custom_output_path)
22
  modelO.config.max_length=512
23
  def summaryA(Text):
24
  tokenized_text = tokenizer.encode(Text,return_tensors="pt")
 
16
 
17
  tokenizer = T5Tokenizer.from_pretrained("AfnanAl/mT5small-ArabicSummary")
18
 
19
+ #model_name="AfnanAl/mT5small-ArabicSummary"
20
+ #custom_output_path="AfnanAl/summaryAonnx"
21
+ modelO=get_onnx_model("AfnanAl/summaryAonnx")
22
  modelO.config.max_length=512
23
  def summaryA(Text):
24
  tokenized_text = tokenizer.encode(Text,return_tensors="pt")