Face-Real-ESRGAN

Running on Zero

doevent commited on Oct 7, 2023

Commit

856b841

•

1 Parent(s): 3be7067

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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