paddle12 commited on
Commit
2eaf83b
·
verified ·
1 Parent(s): ac17754

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ model.eval()
30
  tokenizer = AutoTokenizer.from_pretrained(lora_path)
31
 
32
  # === Updated Generate Answer Function ===
33
- @spaces.GPU(duration=get_duration)
34
  def generate_answer(json_data: Dict[str, Any], question: str) -> str:
35
  """
36
  Generate answer using the fine-tuned model.
 
30
  tokenizer = AutoTokenizer.from_pretrained(lora_path)
31
 
32
  # === Updated Generate Answer Function ===
33
+ @spaces.GPU(duration=60)
34
  def generate_answer(json_data: Dict[str, Any], question: str) -> str:
35
  """
36
  Generate answer using the fine-tuned model.