Update app.py

app.py

@@ -25,12 +25,25 @@ def predict_fn(img):
     values, indices = torch.topk(probabilites, k=5) 
     
     return {LABELS[i]: v.item() for i, v in zip(indices, values)}
+
+# HTML for the header
+header_html = """
+<div style="text-align: center; max-width: 650px; margin: 0 auto;">
+    <h1 style="color: #2c3e50; font-size: 2.5em;">IT Betyár Resnet- Image Classifier</h1>
+    <img src="imgclass.webp" alt="Header Image" style="max-width: 100%; height: auto; margin: 20px 0;">
+    <p style="color: #34495e; font-size: 1.2em;">
+        Üdvözöljük képosztályozónkban! Ez az eszköz egy ImageNeten betanított ResNet50 modellt használja
+        a képek osztályozására. Tölts fel egy képet, és megmutatjuk az 5 legjobb előrejelzést.
+    </p>
+</div>
+"""
     
 gr.Interface(
     predict_fn,
     inputs=gr.components.Image(type='pil'),
     outputs='label',
     title="IT Betyár - ResNet50 Image Classifier",
-    description="Upload an image to get top 5 class predictions.",
+    description="Tölts fel egy képet a predikcióhoz!",
+    article=header_html  # Add the header HTML here
 ).launch() 
-                      
+