Spaces:
Runtime error
Runtime error
Commit
·
97bd2a2
1
Parent(s):
41bfe9d
Update app.py
Browse files
app.py
CHANGED
@@ -32,8 +32,8 @@ def Predict(model, num_images):
|
|
32 |
n += 1
|
33 |
return images
|
34 |
|
35 |
-
def inference(num_images,
|
36 |
-
if
|
37 |
result = Predict(model2, num_images)
|
38 |
else:
|
39 |
result = Predict(model1, num_images)
|
|
|
32 |
n += 1
|
33 |
return images
|
34 |
|
35 |
+
def inference(num_images, Choose: str):
|
36 |
+
if Choose == 'Fashion_mnist':
|
37 |
result = Predict(model2, num_images)
|
38 |
else:
|
39 |
result = Predict(model1, num_images)
|