Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -78,11 +78,12 @@ class FreeAIOrchestrator:
|
|
78 |
# Регистрируем кастомный токенайзер перед загрузкой модели
|
79 |
print("🔄 Инициализация кастомного токенайзера для Surya Table...")
|
80 |
|
81 |
-
# Используем пайплайн с указанием
|
|
|
82 |
self.surya_table_model = pipeline(
|
83 |
"image-to-text",
|
84 |
model="vikp/surya_tablerec",
|
85 |
-
tokenizer=
|
86 |
)
|
87 |
|
88 |
print("✅ Surya Table модель загружена успешно")
|
@@ -264,11 +265,15 @@ class FreeAIOrchestrator:
|
|
264 |
try:
|
265 |
print("🔄 Используем альтернативный метод для Surya Table...")
|
266 |
# Используем автоматически выбранный tokenizer
|
267 |
-
from transformers import pipeline
|
|
|
|
|
|
|
|
|
268 |
surya_fallback = pipeline(
|
269 |
"image-to-text",
|
270 |
model="vikp/surya_tablerec",
|
271 |
-
tokenizer=
|
272 |
)
|
273 |
table_result = surya_fallback(pil_image)
|
274 |
except Exception as fallback_error:
|
@@ -611,6 +616,5 @@ if __name__ == "__main__":
|
|
611 |
app.launch(
|
612 |
server_name="0.0.0.0",
|
613 |
server_port=7860,
|
614 |
-
share=True,
|
615 |
show_error=True
|
616 |
)
|
|
|
78 |
# Регистрируем кастомный токенайзер перед загрузкой модели
|
79 |
print("🔄 Инициализация кастомного токенайзера для Surya Table...")
|
80 |
|
81 |
+
# Используем пайплайн с указанием стандартного токенайзера вместо кастомного
|
82 |
+
# Это решает проблему совместимости
|
83 |
self.surya_table_model = pipeline(
|
84 |
"image-to-text",
|
85 |
model="vikp/surya_tablerec",
|
86 |
+
tokenizer="t5-base" # Используем стандартный токенайзер для t5
|
87 |
)
|
88 |
|
89 |
print("✅ Surya Table модель загружена успешно")
|
|
|
265 |
try:
|
266 |
print("🔄 Используем альтернативный метод для Surya Table...")
|
267 |
# Используем автоматически выбранный tokenizer
|
268 |
+
from transformers import pipeline, AutoTokenizer
|
269 |
+
|
270 |
+
# Загружаем стандартный t5 токенайзер напрямую
|
271 |
+
tokenizer = AutoTokenizer.from_pretrained("t5-base")
|
272 |
+
|
273 |
surya_fallback = pipeline(
|
274 |
"image-to-text",
|
275 |
model="vikp/surya_tablerec",
|
276 |
+
tokenizer=tokenizer # Используем стандартный токенайзер
|
277 |
)
|
278 |
table_result = surya_fallback(pil_image)
|
279 |
except Exception as fallback_error:
|
|
|
616 |
app.launch(
|
617 |
server_name="0.0.0.0",
|
618 |
server_port=7860,
|
|
|
619 |
show_error=True
|
620 |
)
|