Spaces:
Sleeping
Sleeping
Commit
·
50dcc63
1
Parent(s):
eb15969
Fix for Bettertransformer error
Browse files
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 |
-
|
464 |
-
model = BetterTransformer.transform(
|
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
|