Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,10 @@ from vit_model_test import CustomModel
|
|
| 5 |
# Initialize the model
|
| 6 |
model = CustomModel()
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
def predict(image: Image.Image):
|
| 9 |
animation.visible = True # 讛专讗讛 讗转 讛讗谞讬诪爪讬讛
|
| 10 |
label, confidence = model.predict(image)
|
|
|
|
| 5 |
# Initialize the model
|
| 6 |
model = CustomModel()
|
| 7 |
|
| 8 |
+
theme = gr.themes.Base()
|
| 9 |
+
|
| 10 |
+
with gr.Blocks(theme=theme) as demo:
|
| 11 |
+
|
| 12 |
def predict(image: Image.Image):
|
| 13 |
animation.visible = True # 讛专讗讛 讗转 讛讗谞讬诪爪讬讛
|
| 14 |
label, confidence = model.predict(image)
|