Update falcon.py

falcon.py

@@ -33,7 +33,7 @@ def falcon_initial_generation(prompt, instructions, thread):
     job = falcon_client.submit(
         prompt, chathistory, instructions, temperature, p_nucleus_sampling, fn_index=1
     )
-    await wait(job)
+    wait([job])
     if os.environ.get("TEST_ENV") == "True":
         print("falcon text gen job done")
     file_paths = job.outputs()