Update app.py

app.py

@@ -10,12 +10,12 @@ from realesrgan import RealESRGANer
 app = Flask(__name__)
 CORS(app)
 
-# Initialize model with explicit device allocation
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4)
-
-try:
-    upsampler = RealESRGANer(
+# Initialize model
+def init_upsampler():
+    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    model = RRDBNet(num_in_ch=3, num_out_ch=3, num_feat=64, num_block=23, num_grow_ch=32, scale=4)
+    
+    return RealESRGANer(
         scale=4,
         model_path='weights/realesr-general-x4v3.pth',
         model=model,
@@ -25,6 +25,10 @@ try:
         half=False,
         device=device
     )
+
+try:
+    upsampler = init_upsampler()
+    print("Model initialized successfully!")
 except Exception as e:
     print(f"Model initialization failed: {str(e)}")
     upsampler = None
@@ -42,14 +46,8 @@ def enhance_image():
         img = Image.open(file.stream).convert('RGB')
         img_array = np.array(img)
         
-        # Process image
-        output, _ = upsampler.enhance(
-            img_array,
-            outscale=4,
-            alpha_upsampler='realesrgan'
-        )
+        output, _ = upsampler.enhance(img_array, outscale=4)
         
-        # Convert to bytes
         img_byte_arr = io.BytesIO()
         Image.fromarray(output).save(img_byte_arr, format='JPEG')
         img_byte_arr.seek(0)
@@ -61,7 +59,7 @@ def enhance_image():
 
 @app.route('/health', methods=['GET'])
 def health_check():
-    status = 'ready' if upsampler else 'initializing'
+    status = 'ready' if upsampler else 'unavailable'
     return {'status': status}, 200
 
 if __name__ == '__main__':