Upscale

Runtime error

doevent commited on Mar 30, 2023

Commit

bdfa160

•

1 Parent(s): 0969366

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,8 +19,8 @@ def inference(image: Image, size: str) -> Image:
         result = model4.predict(image.convert('RGB'))
     else:
         result = model8.predict(image.convert('RGB'))
-    # if torch.cuda.is_available():
-    #     torch.cuda.empty_cache()
     return result

         result = model4.predict(image.convert('RGB'))
     else:
         result = model8.predict(image.convert('RGB'))
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
     return result