Update app.py

app.py

@@ -1,6 +1,5 @@
 import numpy as np
 import gradio as gr
-# teo
 import torch
 from transformers import Dinov2Config, Dinov2Model, Dinov2ForImageClassification, AutoImageProcessor
 import torch.nn as nn
@@ -27,6 +26,7 @@ class NewheadDinov2ForImageClassification(Dinov2ForImageClassification):
 
         # Classifier head
         self.classifier = create_head(config.hidden_size * 2, config.num_labels)
+        
 # IMPORT CLASSIFICATION MODEL
 checkpoint_name = "lombardata/dino-base-2023_11_27-with_custom_head"
 # import labels
@@ -63,12 +63,20 @@ def predict(input_image):
         i += 1
     result = {key: result[key] for key in result if result[key] > 0.5}
     return result
+    
+# Define style
+title = "DinoVd'eau image classification"
+description = f"This is a prototype application that demonstrates how artificial intelligence-based systems can recognize what object(s) 
+is present in an underwater image. To use it, simply upload your image, or click one of the example images to load them. For predictions, 
+we use the open-source model {checkpoint_name}"
 
 gr.Interface(
     fn=predict,
     inputs=gr.Image(shape=(224, 224)),
     #outputs=gr.Label(num_top_classes=5),
-    outputs="label").launch()
+    outputs="label",
+    title=title, 
+    description=description).launch()
 
     
     #examples=["GOPR0106.JPG",