Update app.py

app.py

@@ -8,8 +8,8 @@ import uuid
 
 
 # Load model and tokenizer
-our_model_path = "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B"
-#our_model_path = "FractalAIResearch/Fathom-R1-14B"
+#our_model_path = "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B"
+our_model_path = "FractalAIResearch/Fathom-R1-14B"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 our_model = AutoModelForCausalLM.from_pretrained(our_model_path, trust_remote_code=True,device_map="auto", torch_dtype="auto")
@@ -26,7 +26,7 @@ conversations = {}
 def generate_conversation_id():
     return str(uuid.uuid4())[:8]
 
-@spaces.GPU(duration=60)
+@spaces.GPU(duration=600)
 def generate_response(user_message, max_tokens, temperature, top_p, history_state):
     if not user_message.strip():
         return history_state, history_state