Shekswess commited on
Commit
f6270b7
·
1 Parent(s): bfef1ae

Removing stop option

Browse files
Files changed (1) hide show
  1. app.py +0 -21
app.py CHANGED
@@ -2,7 +2,6 @@
2
 
3
  import os
4
  import asyncio
5
- from huggingface_hub import HfFolder
6
 
7
  import gradio as gr
8
  from synthgenai import DatasetConfig, DatasetGeneratorConfig, LLMConfig, InstructionDatasetGenerator, PreferenceDatasetGenerator,RawDatasetGenerator,SentimentAnalysisDatasetGenerator, SummarizationDatasetGenerator, TextClassificationDatasetGenerator
@@ -143,14 +142,8 @@ def generate_synthetic_dataset(
143
  else:
144
  return "Invalid dataset type"
145
 
146
- stop_event.clear()
147
-
148
  async def generate():
149
- if stop_event.is_set():
150
- return "Dataset generation stopped."
151
  dataset = await generator.agenerate_dataset()
152
- if stop_event.is_set():
153
- return "Dataset generation stopped."
154
  dataset.save_dataset(hf_repo_name=hf_repo_name)
155
  return "Dataset generated and saved successfully."
156
 
@@ -244,20 +237,11 @@ def ui_main():
244
  )
245
 
246
  generate_button = gr.Button("Generate Dataset")
247
- stop_button = gr.Button("Stop Generation", visible=False)
248
  output = gr.Textbox(label="Operation Result", value="")
249
 
250
  def on_generate_click(*args):
251
- generate_button.visible = False
252
- stop_button.visible = True
253
  return generate_synthetic_dataset(*args)
254
 
255
- def on_stop_click():
256
- stop_generation()
257
- generate_button.visible = True
258
- stop_button.visible = False
259
- return "Dataset generation stopped."
260
-
261
  generate_button.click(
262
  on_generate_click,
263
  inputs=[
@@ -278,11 +262,6 @@ def ui_main():
278
  outputs=output,
279
  )
280
 
281
- stop_button.click(
282
- on_stop_click,
283
- outputs=output,
284
- )
285
-
286
  demo.launch(inbrowser=True, favicon_path=None)
287
 
288
 
 
2
 
3
  import os
4
  import asyncio
 
5
 
6
  import gradio as gr
7
  from synthgenai import DatasetConfig, DatasetGeneratorConfig, LLMConfig, InstructionDatasetGenerator, PreferenceDatasetGenerator,RawDatasetGenerator,SentimentAnalysisDatasetGenerator, SummarizationDatasetGenerator, TextClassificationDatasetGenerator
 
142
  else:
143
  return "Invalid dataset type"
144
 
 
 
145
  async def generate():
 
 
146
  dataset = await generator.agenerate_dataset()
 
 
147
  dataset.save_dataset(hf_repo_name=hf_repo_name)
148
  return "Dataset generated and saved successfully."
149
 
 
237
  )
238
 
239
  generate_button = gr.Button("Generate Dataset")
 
240
  output = gr.Textbox(label="Operation Result", value="")
241
 
242
  def on_generate_click(*args):
 
 
243
  return generate_synthetic_dataset(*args)
244
 
 
 
 
 
 
 
245
  generate_button.click(
246
  on_generate_click,
247
  inputs=[
 
262
  outputs=output,
263
  )
264
 
 
 
 
 
 
265
  demo.launch(inbrowser=True, favicon_path=None)
266
 
267