Update app.py

app.py

@@ -10,9 +10,12 @@ model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
 
 # Функция для генерации текста
 def generate_text(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(**inputs, max_length=100)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+    try:
+        inputs = tokenizer(prompt, return_tensors="pt")
+        outputs = model.generate(**inputs, max_length=100)
+        return tokenizer.decode(outputs[0], skip_special_tokens=True)
+    except Exception as e:
+        return f"Ошибка при генерации текста: {str(e)}"
 
 # Gradio интерфейс
 interface = gr.Interface(
@@ -22,5 +25,4 @@ interface = gr.Interface(
 )
 
 # Запуск
-interface.launch()
-
+interface.launch()