bertugmirasyedi commited on
Commit
50dcc63
·
1 Parent(s): eb15969

Fix for Bettertransformer error

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -460,8 +460,8 @@ async def summarize(descriptions: list, runtime="normal"):
460
  # Define the summarizer model and tokenizer
461
  if runtime == "normal":
462
  tokenizer = summary_tokenizer_normal
463
- model = summary_model_normal
464
- model = BetterTransformer.transform(model)
465
  elif runtime == "onnxruntime":
466
  tokenizer = summary_tokenizer_onnx
467
  model = summary_model_onnx
 
460
  # Define the summarizer model and tokenizer
461
  if runtime == "normal":
462
  tokenizer = summary_tokenizer_normal
463
+ normal_model = summary_model_normal
464
+ model = BetterTransformer.transform(normal_model)
465
  elif runtime == "onnxruntime":
466
  tokenizer = summary_tokenizer_onnx
467
  model = summary_model_onnx