Update app.py
Browse files
app.py
CHANGED
@@ -283,7 +283,7 @@ def update_json(complex_name, protein_chains, dna_sequences, rna_sequences, liga
|
|
283 |
|
284 |
|
285 |
#@torch.inference_mode()
|
286 |
-
@spaces.GPU(duration=
|
287 |
def predict_structure(input_collector: dict):
|
288 |
#first initialize runner
|
289 |
runner = InferenceRunner(configs)
|
|
|
283 |
|
284 |
|
285 |
#@torch.inference_mode()
|
286 |
+
@spaces.GPU(duration=180) # Specify a duration to avoid timeout
|
287 |
def predict_structure(input_collector: dict):
|
288 |
#first initialize runner
|
289 |
runner = InferenceRunner(configs)
|