Update app.py

app.py

@@ -1,9 +1,11 @@
 import torch
 import torchvision.transforms as transforms
-import gadio as gr
+import gradio as gr
 from huggingface_hub import hf_hub_download
 from PIL import Image
 
+print("Stage 0: Completed")
+
 Fruits = ['Acerola', 'Apple', 'Apricot', 'Avocado', 'Banana', 'Black Berry', 'Blue Berry', 'Cantaloupe', 'Cherry',
           'Coconut', 'Fig', 'Grapefruit', 'Grape', 'Guava', 'Kiwi Fruit', 'Lemon', 'Lime', 'Mango', 'Olive', 'Orange',
           'Passion Fruit', 'Peach', 'Pear', 'Pineapple', 'Plum', 'Pomegranate', 'Raspberry', 'Strawberry', 'Tomato',
@@ -17,6 +19,8 @@ model_path = hf_hub_download(repo_id = repo_name, file_name = file_name)
 model = torch.load(model_path).to(device)
 model.eval()
 
+print("Stage 1: Completed ")
+
 transform = transforms.Compose([
     transforms.Resize(224),
     transforms.CenterCrop(224),
@@ -39,4 +43,6 @@ interface = gr.Interface(
     theme = gr.themes.soft()
 ) 
 
+print("Stage 2: Completed")
+
 interface.launch(debug = True)