Update app.py

app.py

@@ -6,8 +6,8 @@ from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 
 dataset = load_dataset("beans")
 
-extractor = AutoFeatureExtractor.from_pretrained("nateraw/vit-base-beans")
-model = AutoModelForImageClassification.from_pretrained("nateraw/vit-base-beans")
+extractor = AutoFeatureExtractor.from_pretrained("susnato/plant_disease_detection-beans")
+model = AutoModelForImageClassification.from_pretrained("susnato/plant_disease_detection-beans")
 
 labels = ['angular_leaf_spot', 'rust', 'healthy']
 
@@ -20,7 +20,7 @@ def classify(im):
   return confidences
 
 
-block = gr.Blocks(theme="gary109/HaleyCH_Theme")
+block = gr.Blocks(theme="JohnSmith9982/small_and_pretty")
 with block:
     gr.HTML(
         """
@@ -33,9 +33,8 @@ with block:
                 """
                 <p style="color:black">
                     <h4 style="font-color:powderblue;">
-                        <center>Crop diseases are a major threat to food security, but their rapid identification remains difficult in many parts of the world due to the lack of the necessary infrastructure. The combination of increasing global smartphone penetration and recent advances in computer vision made possible by deep learning has paved the way for smartphone-assisted disease diagnosis.</center>
-                        <br>
-                        <center>Using A.I. models in plant disease detection and diagnosis has the potential to revolutionize the way we approach agriculture. By providing real-time monitoring and accurate detection of plant diseases, A.I. can help farmers reduce costs and increase crop</center>
+                        <center>Crop diseases are a major threat to food security, but their rapid identification remains difficult in many parts of the world due to the lack of the necessary infrastructure. <br><br>
+                        Using Computer Vision models in plant disease detection and diagnosis has the potential to revolutionize the way we approach agriculture. By providing real-time monitoring and accurate detection of plant diseases, A.I. can help farmers reduce costs and increase crop</center>
                     </h4>
                 </p>
                 
@@ -45,6 +44,18 @@ with block:
                 """
             )
 
+    with gr.Group():
+        with gr.Row():
+            gr.HTML(
+                """
+                <center><h3>Our Approach</h3></center>
+                
+                <p align="center">
+                    <img src="https://huggingface.co/datasets/susnato/stock_images/resolve/main/diagram2.png">
+                </p>
+                """
+            )
+
     with gr.Group():
         image = gr.Image(type='pil')
         outputs = gr.Label()
@@ -56,9 +67,7 @@ with block:
                      )
 
     with gr.Group():
-        gr.Examples([
-            ["ex1.jpg", "ex3.jpg"],
-        ],
+        gr.Examples(["ex3.jpg"],
             fn=classify,
             inputs=[image],
             outputs=[outputs],