Update inference.py

inference.py

@@ -6,6 +6,10 @@ def load_model():
     return model
 
 def predict(model, inputs):
+    for itr, input in enumerate(inputs):
+        input = torch.tensor(input)
+    
     with torch.no_grad():
-        outputs = model(inputs)
-    return outputs.numpy().tolist()
+        output = model(*inputs)
+    
+    return output.numpy().tolist()