app.py

app.py

@@ -0,0 +1,25 @@
+import gradio as gr
+from transformers import pipeline
+
+# Wczytywanie modelu Bielik-7B
+model = pipeline("text-generation", model="speakleash/Bielik-7B-v0.1")
+
+def generate_description(product_name, product_features):
+    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']
+
+# Konfiguracja interfejsu Gradio
+iface = gr.Interface(
+    fn=generate_description,
+    inputs=["text", "text"],
+    outputs="text",
+    title="Bielik-7B Product Description Generator",
+    description="Wprowadź nazwę produktu i jego cechy, aby wygenerować opis."
+)
+
+if __name__ == "__main__":
+    iface.launch()