Shekswess commited on
Commit
e36771a
·
1 Parent(s): e5e6a27

Event loop close

Browse files
Files changed (1) hide show
  1. app.py +14 -3
app.py CHANGED
@@ -84,9 +84,20 @@ def generate_synthetic_dataset(
84
  else:
85
  return "Invalid dataset type"
86
 
87
- dataset = asyncio.run(generator.agenerate_dataset())
88
- dataset.save_dataset(hf_repo_name=hf_repo_name)
89
- return "Dataset generated and saved successfully."
 
 
 
 
 
 
 
 
 
 
 
90
 
91
  def ui_main():
92
  """
 
84
  else:
85
  return "Invalid dataset type"
86
 
87
+ async def generate():
88
+ dataset = await generator.agenerate_dataset()
89
+ dataset.save_dataset(hf_repo_name=hf_repo_name)
90
+ return "Dataset generated and saved successfully."
91
+
92
+ try:
93
+ return asyncio.run(generate())
94
+ except RuntimeError as e:
95
+ if str(e) == "Event loop is closed":
96
+ loop = asyncio.new_event_loop()
97
+ asyncio.set_event_loop(loop)
98
+ return loop.run_until_complete(generate())
99
+ else:
100
+ raise
101
 
102
  def ui_main():
103
  """