Update app.py

app.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 import os
+import glob  # Added missing import
 import streamlit as st
 from PIL import Image
 import torch
@@ -72,8 +73,7 @@ def load_image_gen():
     return pipeline
 
 def load_line_drawer():
-    # Simplified from your Torch Space (assuming a UNet-like model for edge detection)
-    # Placeholder: Using OpenCV edge detection as a minimal CPU example
+    # Simplified OpenCV-based edge detection (CPU-friendly substitute for Torch Space UNet)
     def edge_detection(image):
         img_np = np.array(image.convert("RGB"))
         gray = cv2.cvtColor(img_np, cv2.COLOR_RGB2GRAY)