Update app.py

app.py

@@ -7,10 +7,10 @@ import os
 import requests
 import hashlib
 
-# Configuration
+# Configuration - UPDATED MODEL URL
 MODEL_URL = "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.onnx"
 MODEL_PATH = "realesr.onnx"
-EXPECTED_MD5 = "8a628e89b1e4d9f5f174a3e8c0c7b3b1"  # MD5 hash of correct model file
+EXPECTED_MD5 = "8a628e89b1e4d9f5f174a3e8c0c7b3b1"  # MD5 hash for verification
 
 def verify_file(file_path, expected_md5):
     """Verify file integrity using MD5 hash"""
@@ -21,25 +21,39 @@ def verify_file(file_path, expected_md5):
     return file_hash == expected_md5
 
 def download_model():
-    """Download model with verification"""
+    """Download model with verification and retries"""
     print("Downloading model...")
-    try:
-        response = requests.get(MODEL_URL, stream=True)
-        response.raise_for_status()
-        
-        # Save in chunks to handle large files
-        with open(MODEL_PATH, "wb") as f:
-            for chunk in response.iter_content(chunk_size=8192):
-                f.write(chunk)
-        
-        # Verify download
-        if not verify_file(MODEL_PATH, EXPECTED_MD5):
-            os.remove(MODEL_PATH)
-            raise ValueError("Downloaded file is corrupted")
+    for attempt in range(3):  # Retry up to 3 times
+        try:
+            # Use a mirror URL if primary fails
+            urls = [
+                MODEL_URL,
+                "https://huggingface.co/spaces/akhaliq/Real-ESRGAN/resolve/main/realesr-general-x4v3.onnx"
+            ]
             
-        print("Model downloaded and verified successfully!")
-    except Exception as e:
-        raise gr.Error(f"Model download failed: {str(e)}")
+            for url in urls:
+                try:
+                    response = requests.get(url, stream=True, timeout=30)
+                    response.raise_for_status()
+                    
+                    # Save in chunks
+                    with open(MODEL_PATH, "wb") as f:
+                        for chunk in response.iter_content(chunk_size=8192):
+                            f.write(chunk)
+                    
+                    # Verify download
+                    if verify_file(MODEL_PATH, EXPECTED_MD5):
+                        print("Model downloaded and verified successfully!")
+                        return
+                    os.remove(MODEL_PATH)
+                except Exception:
+                    continue
+            
+            raise ValueError("All download attempts failed")
+        except Exception as e:
+            if attempt == 2:  # Final attempt
+                raise gr.Error(f"Model download failed after 3 attempts: {str(e)}")
+            print(f"Attempt {attempt + 1} failed, retrying...")
 
 # Download model if missing or corrupted
 if not verify_file(MODEL_PATH, EXPECTED_MD5):
@@ -53,7 +67,7 @@ try:
     )
     print("Model loaded successfully!")
 except Exception as e:
-    raise gr.Error(f"Model loading failed: {str(e)}\nTry deleting and reuploading the model file.")
+    raise gr.Error(f"Model loading failed: {str(e)}\nTry deleting the space and recreating it.")
 
 def enhance(image):
     """Image enhancement function"""