Update app.py

app.py

@@ -13,16 +13,12 @@ model8.load_weights('weights/RealESRGAN_x8.pth', download=True)
 
 
 def inference(image: Image, size: str) -> Image:
-    try:
-        if size == '2x':
-            result = model2.predict(image.convert('RGB'))
-        elif size == '4x':
-            result = model4.predict(image.convert('RGB'))
-        else:
-            result = model8.predict(image.convert('RGB'))
-    except Exception as e:
-        print(e)
-        raise gr.Error(str(e))
+    if size == '2x':
+        result = model2.predict(image.convert('RGB'))
+    elif size == '4x':
+        result = model4.predict(image.convert('RGB'))
+    else:
+        result = model8.predict(image.convert('RGB'))
     return result