Spaces:
Running
Running
Removing stop option
Browse files
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 |
|