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
|