Update app.py

app.py

@@ -1,3 +1,14 @@
 import gradio as gr
 
-gr.Interface.load("models/google/vit-base-patch16-224").launch()
+# Instead of launching the model directly, save it to a variable
+model_interface = gr.Interface.load("models/google/vit-base-patch16-224")
+
+# Customize the interface with the desired template properties
+model_interface.title = "Image classification"
+model_interface.description = "Upload an image here and classify it based on Google models"
+model_interface.theme = "Monochrome"
+model_interface.live = True
+model_interface.capture_session = True
+
+# Launch the customized Gradio interface
+model_interface.launch()