app.py
CHANGED
|
@@ -68,6 +68,8 @@ def greet(input_img):
|
|
| 68 |
outputs = model(**inputs)
|
| 69 |
logits = outputs.logits
|
| 70 |
|
|
|
|
|
|
|
| 71 |
logits = tf.transpose(logits, [0, 2, 3, 1])
|
| 72 |
logits = tf.image.resize(logits, input_img.size[::-1])
|
| 73 |
|
|
|
|
| 68 |
outputs = model(**inputs)
|
| 69 |
logits = outputs.logits
|
| 70 |
|
| 71 |
+
logits = logits.detach().numpy()
|
| 72 |
+
|
| 73 |
logits = tf.transpose(logits, [0, 2, 3, 1])
|
| 74 |
logits = tf.image.resize(logits, input_img.size[::-1])
|
| 75 |
|