Update app.py

app.py

@@ -16,6 +16,9 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 MAX_LENGTH_REQUEST = 1024
 MAX_NEW_TOKENS = 128
 MAX_LENGTH_RESPONSE = 100
+TEST_ENV=os.getenv("TEST_ENV")
+
+logger.info(f"TEST_ENV= {TEST_ENV}")
 
 # Настройка логирования
 logging.basicConfig(
@@ -40,7 +43,7 @@ sber_rugpt3small_based_on_gpt2_model_name = "sberbank-ai/rugpt3small_based_on_gp
 try:
     model_name = rugpt3large_based_on_gpt2_model_name  # Меньшая модель
     tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="left")
-    model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", from_flax=True)
+    model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
     logger.info("Модель успешно загружена")
 except Exception as e:
     logger.error(f"Ошибка загрузки модели: {str(e)}")