AfnanAl commited on
Commit
c62a8ee
·
1 Parent(s): 7f82037

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,10 +16,10 @@ 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 = quantize("AfnanAl/summaryAonnx")
22
- modelO=get_onnx_model(model_name, custom_output_path)
23
  modelO.config.max_length=512
24
  def summaryA(Text):
25
  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 = quantize("AfnanAl/summaryAonnx")
22
+ modelO=get_onnx_model(custom_output_path)
23
  modelO.config.max_length=512
24
  def summaryA(Text):
25
  tokenized_text = tokenizer.encode(Text,return_tensors="pt")