Update app.py

app.py

@@ -141,7 +141,7 @@ def generate_response(instruction, model_path, progress=gr.Progress()):
     except Exception as e:
         return f"Error generating response: {str(e)}"
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=200)
 def judge_responses(instruction, response1, response2, model_name, temperature=0.1, max_new_tokens=2048):
     """
     Evaluate the quality of two responses
@@ -216,7 +216,7 @@ def judge_responses(instruction, response1, response2, model_name, temperature=0
             
         yield result
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=200)
 def generate_and_judge(instruction, model_dropdown_1, custom_model_1, model_dropdown_2, custom_model_2, judge_model_name, temperature=0.1, max_new_tokens=2048, progress=gr.Progress()):
     """Generate responses from two models and judge them"""
     progress(0, desc="Starting generation process")