Update app.py

app.py

@@ -1,16 +1,30 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, set_seed
 
-# Wczytywanie modelu Bielik-7B
-model = pipeline("text-generation", model="speakleash/Bielik-7B-v0.1")
+# Funkcja do ładowania modelu
+def load_model():
+    try:
+        model = pipeline("text-generation", model="speakleash/Bielik-7B-v0.1")
+        return model
+    except Exception as e:
+        print(f"Error loading model: {e}")
+        return None
+
+model = load_model()
 
 def generate_description(product_name, product_features):
+    if model is None:
+        return "Model not loaded correctly."
+
     prompt = (
         f"Opisz produkt o nazwie {product_name}, który posiada następujące cechy: {product_features}. "
         "Podaj jego główne zalety, zastosowania i cechy wyróżniające. Wyjaśnij, dlaczego warto go kupić."
     )
-    result = model(prompt, max_length=200, temperature=0.7, top_p=0.9)
-    return result[0]['generated_text']
+    try:
+        result = model(prompt, max_length=200, temperature=0.7, top_p=0.9)
+        return result[0]['generated_text']
+    except Exception as e:
+        return f"Error generating description: {e}"
 
 # Konfiguracja interfejsu Gradio
 iface = gr.Interface(