Lyti4 commited on
Commit
c99d028
·
verified ·
1 Parent(s): cded211

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -5
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=Byt5LangTokenizer.from_pretrained("t5-base", legacy=False)
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=None # Позволяем pipeline самому выбрать 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
  )