Spaces:
Build error
Build error
Commit
β’
c70f662
1
Parent(s):
da12076
Label Changes
Browse files
app.py
CHANGED
@@ -97,7 +97,11 @@ def classify_info(image_1, text_1, image_2, text_2):
|
|
97 |
ds = ds.batch(batch_size).prefetch(auto)
|
98 |
output = model.predict(ds)
|
99 |
|
100 |
-
outputs =
|
|
|
|
|
|
|
|
|
101 |
#label = np.argmax(output)
|
102 |
return outputs #labels[label]
|
103 |
|
|
|
97 |
ds = ds.batch(batch_size).prefetch(auto)
|
98 |
output = model.predict(ds)
|
99 |
|
100 |
+
outputs = dict()
|
101 |
+
|
102 |
+
outputs[labels[0]] = float(output[0][0])
|
103 |
+
outputs[labels[1]] = float(output[0][1])
|
104 |
+
outputs[labels[2]] = float(output[0][2])
|
105 |
#label = np.argmax(output)
|
106 |
return outputs #labels[label]
|
107 |
|