upscale

Running

doevent commited on Oct 14, 2023

Commit

1f67e13

•

1 Parent(s): d8aabe2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,6 +20,8 @@ def inference(image, size):
         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