Update app.py

app.py

@@ -4,17 +4,16 @@ import gradio as gr
 import numpy as np
 from huggingface_hub.fastai_utils import from_pretrained_fastai
 from fastai.vision.all import *
-import matplotlib.pyplot as plt
 from pathlib import Path
 
 # ================================================
 # STUBS FOR CUSTOM CLASSES / FUNCTIONS
-# These match all custom classes used in the model
 # ================================================
 
 # Functions
 def _inner(*args, **kwargs): return None
 def style_loss(*args, **kwargs): return None
+def act_loss(*args, **kwargs): return None
 
 # Layers
 class ReflectionLayer(nn.Module):
@@ -52,7 +51,7 @@ learner = from_pretrained_fastai(REPO_ID)
 # ================================================
 
 def infer(img):
-    # Ensure PILImage
+    # Convert to PILImage if needed
     if not isinstance(img, PILImage):
         img = PILImage.create(img)