Merge pull request #38 from aurotripathy/fix-quick-start-example

changed input range from int8 to uint8
furiosa-ai · Jan 13, 2023 · 7a30eb6 · 7a30eb6
2 parents 9379820 + dbaa0e0
commit 7a30eb6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/inferences/quickstart_example.py b/examples/inferences/quickstart_example.py
@@ -26,10 +26,10 @@ def run_example():
   print(input_meta)
 
   # Generate the random input tensor according to the input shape
-  input = np.random.randint(-128, 127, input_meta.shape(), dtype=np.int8)
+  input = np.random.randint(0, 255, input_meta.shape, dtype=np.uint8)
 
   # Run the inference
-  outputs = sess.run(input)
+  outputs = sess.run(input.astype(np.uint8))
 
   print("== Output ==")
   print(outputs)