Upscale

Runtime error

doevent commited on Jan 30, 2023

Commit

357cf6f

•

1 Parent(s): e198db2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,6 +19,8 @@ def inference(image: Image, size: str) -> Image:
         result = model4.predict(image.convert('RGB'))
     else:
         result = model8.predict(image.convert('RGB'))
     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