f

app.py

@@ -31,7 +31,7 @@ def askcpv(description):
     print("probs: ", probs.detach().numpy())
     print("values: ", values.detach().numpy())
     print("indices: ", indices.detach().numpy())
-    print({i: v.item() for i, v in zip(indices, values)})
-    return {cpv[i]: v.item() for i, v in zip(indices, values)}
+    # print({i: v.item() for i, v in zip(indices, values)})
+    return {cpv[i]: v.item() for i, v in zip(indices.detach().numpy(), values.detach().numpy())}
 
 gr.Interface(fn=askcpv, inputs="textbox", outputs="label").launch()