Update app.py

app.py

@@ -10,7 +10,7 @@ from main import generate_data, PROMPT_1
 from topics import TOPICS
 from system_messages import SYSTEM_MESSAGES_VODALUS
 import random
-import spaces
+
 
 ANNOTATION_CONFIG_FILE = "annotation_config.json"
 OUTPUT_FILE_PATH = "dataset.jsonl"
@@ -264,7 +264,7 @@ def save_dataset_config(system_messages, prompt_1, topics):
     
     return "Dataset configuration saved successfully"
 
-@spaces.GPU(duration=120)
+
 def chat_with_llm(message, history):
     try:
         msg_list = [{"role": "system", "content": "You are an AI assistant helping with dataset annotation and quality checking."}]
@@ -292,7 +292,7 @@ def update_chat_context(row_data, index, total, quality, high_quality_tags, low_
     """
     return [[None, context]]  # Return as a list of message pairs
 
-#@spaces.GPU(duration=180)
+
 async def run_generate_dataset(num_workers, num_generations, output_file_path):
     generated_data = []
     for _ in range(num_generations):