Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ else:
|
|
42 |
predictions = cnn.apply({"params": params}, input)
|
43 |
for (image, prediction) in zip(uploaded_files, predictions):
|
44 |
st.image(Image.open(image))
|
45 |
-
st.write("Model Prediction:
|
46 |
|
47 |
|
48 |
def gridify(kernel, grid, kernel_size, scaling=5, padding=1):
|
|
|
42 |
predictions = cnn.apply({"params": params}, input)
|
43 |
for (image, prediction) in zip(uploaded_files, predictions):
|
44 |
st.image(Image.open(image))
|
45 |
+
st.write(f"Model Prediction: {prediction.argmax().tolist()}")
|
46 |
|
47 |
|
48 |
def gridify(kernel, grid, kernel_size, scaling=5, padding=1):
|