Update app.py

app.py

@@ -1,3 +1,19 @@
+import sys
+import subprocess
+
+# Try to import OpenCV; if not available, install it
+try:
+    import cv2
+    print("OpenCV already installed.")
+except ImportError:
+    print("OpenCV not found. Installing opencv-python-headless...")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "opencv-python-headless"])
+    try:
+        import cv2
+        print("OpenCV successfully installed.")
+    except ImportError:
+        print("Failed to install OpenCV. Continuing without it...")
+
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer