Update app.py

app.py

@@ -19,9 +19,6 @@ def inference(image, size):
     if image is None:
         raise gr.Error("Image not uploaded")
         
-    width, height = image.size
-    if width >= 5000 or height >= 5000:
-        raise gr.Error("The image is too large.")
 
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
@@ -44,6 +41,9 @@ def inference(image, size):
             result = model2.predict(image.convert('RGB'))
     else:
         try:
+            width, height = image.size
+            if width >= 5000 or height >= 5000:
+                raise gr.Error("The image is too large.")
             result = model8.predict(image.convert('RGB'))
         except torch.cuda.OutOfMemoryError as e:
             print(e)