Update run.py

run.py

@@ -76,13 +76,9 @@ if __name__ == "__main__":
     data = {"id": 0, "question": "What is the capital of France?"}  # This can be a list of samples
     # data = {"id": 0, "question": "Who was the NBA champion in 2023?"}  # This can be a list of samples
     
-    #option1: use the FlowMessage class
-    input_message = FlowMessage(
-        data=data,
-    )
-
+ 
     #option2: use the proxy_flow
-    #input_message = proxy_flow.package_input_message(data = data)
+    input_message = proxy_flow.package_input_message(data = data)
     
     #7. ~~~ Run inference ~~~
     future = proxy_flow.get_reply_future(input_message)