Update app.py

app.py

@@ -106,10 +106,14 @@ model.load_state_dict(torch.load('best_model_nikita.pt', map_location=device))
 # 4. Set model to evaluation mode
 model.eval()
 
+title = "BaMI Classifier"
+description = "A plant based classifier to classify the particular species of Mushroom and Berries, along with it's edibality"
 
 demo = gr.Interface(fn = predict,
              inputs=gr.inputs.Image(type="pil"),
              outputs=[gr.outputs.Label(num_top_classes=1), gr.outputs.Label(num_top_classes=3)], # selec
+             examples=['image3.jpg', 'image9.jpg', 'image12.jpg'],
+             title=title,description=description
              )
 
 demo.launch(debug=True)