Add error handling for Zero GPU initialization and fallback to CPU mode; update Gradio version in requirements

app.py

@@ -33,6 +33,11 @@ try:
 except ImportError:
     SPACES_AVAILABLE = False
     print("Running without Zero GPU support")
+except Exception as e:
+    # Catch any other initialization errors
+    SPACES_AVAILABLE = False
+    print(f"Zero GPU initialization warning: {e}")
+    print("Running without Zero GPU support")
 
 # Keep-alive state
 last_activity = datetime.datetime.now()
@@ -388,7 +393,12 @@ def score_and_create_matrix_all_singles_impl(sequence,mutation_range_start=None,
 
 # Apply Zero GPU decorator if available
 if SPACES_AVAILABLE:
-    score_and_create_matrix_all_singles = spaces.GPU(duration=300)(score_and_create_matrix_all_singles_impl)
+    try:
+        score_and_create_matrix_all_singles = spaces.GPU(duration=300)(score_and_create_matrix_all_singles_impl)
+    except Exception as e:
+        print(f"Warning: Could not apply Zero GPU decorator: {e}")
+        print("Falling back to CPU mode")
+        score_and_create_matrix_all_singles = score_and_create_matrix_all_singles_impl
 else:
     score_and_create_matrix_all_singles = score_and_create_matrix_all_singles_impl
 
@@ -536,9 +546,10 @@ if __name__ == "__main__":
     print("Note: Models will be downloaded on first use")
     print("Zero GPU spaces may sleep after ~15 minutes of inactivity")
     
-    # Simple launch without queue to avoid Zero GPU conflicts
-    tranception_design.launch(
+    # Launch with queue for proper Zero GPU support
+    tranception_design.queue(max_size=20).launch(
         server_name="0.0.0.0",
         server_port=7860,
-        show_error=True
+        show_error=True,
+        share=False
     )

requirements.txt

@@ -10,7 +10,7 @@ tqdm
 biopython>=1.78,<2.0.0
 matplotlib
 seaborn
-gradio==3.50.0
+gradio>=4.0.0,<5.0.0
 datasets==1.18.3
 wandb==0.12.11
 jax>=0.3.25