Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
| 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.
|