Spaces:
Runtime error
Runtime error
333
Browse files
app.py
CHANGED
|
@@ -14,7 +14,6 @@ def greet(img):
|
|
| 14 |
|
| 15 |
def process_sketch(sketch):
|
| 16 |
img = np.expand_dims(img, axis=0)
|
| 17 |
-
sketch = np.array(sketch)
|
| 18 |
output1 = np.argmax(model.predict(img)[0])
|
| 19 |
output2 = (np.argmax(model.predict(img)[0]) % 3)
|
| 20 |
return output1, output2
|
|
|
|
| 14 |
|
| 15 |
def process_sketch(sketch):
|
| 16 |
img = np.expand_dims(img, axis=0)
|
|
|
|
| 17 |
output1 = np.argmax(model.predict(img)[0])
|
| 18 |
output2 = (np.argmax(model.predict(img)[0]) % 3)
|
| 19 |
return output1, output2
|