Spaces:
Runtime error
Runtime error
Upload 5 files
Browse files- model.py +1 -1
- requirements.txt +1 -1
model.py
CHANGED
|
@@ -21,4 +21,4 @@ def create_effnetb2_model(num_classes:int=3, # default output classes = 3 (pizza
|
|
| 21 |
nn.Linear(in_features=1408, out_features=num_classes)
|
| 22 |
)
|
| 23 |
|
| 24 |
-
return model, transforms
|
|
|
|
| 21 |
nn.Linear(in_features=1408, out_features=num_classes)
|
| 22 |
)
|
| 23 |
|
| 24 |
+
return model, transforms
|
requirements.txt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
torch
|
| 2 |
torchvision
|
| 3 |
-
gradio==3.1.4
|
|
|
|
| 1 |
torch
|
| 2 |
torchvision
|
| 3 |
+
gradio==3.1.4
|