Spaces:
Sleeping
Sleeping
Update difpoint/src/models/predictor.py
Browse files
difpoint/src/models/predictor.py
CHANGED
@@ -226,8 +226,6 @@ class OnnxRuntimePredictor:
|
|
226 |
input_feeds[self.inputs[i].name] = data[i].astype(np.float16)
|
227 |
else:
|
228 |
input_feeds[self.inputs[i].name] = data[i].astype(np.float32)
|
229 |
-
for key, value in input_feeds.items():
|
230 |
-
print(f"Input: {key}, Shape: {value.shape}")
|
231 |
results = self.onnx_model.run(None, input_feeds)
|
232 |
return results
|
233 |
|
|
|
226 |
input_feeds[self.inputs[i].name] = data[i].astype(np.float16)
|
227 |
else:
|
228 |
input_feeds[self.inputs[i].name] = data[i].astype(np.float32)
|
|
|
|
|
229 |
results = self.onnx_model.run(None, input_feeds)
|
230 |
return results
|
231 |
|