more fixes

app.py

@@ -1,32 +1,6 @@
-import torch
-import sys
-import subprocess
-
-def check_gpu_status():
-    print("Python version:", sys.version)
-    print("PyTorch version:", torch.__version__)
-    print("CUDA available:", torch.cuda.is_available())
-    print("CUDA version:", torch.version.cuda if torch.cuda.is_available() else "Not available")
-    
-    if torch.cuda.is_available():
-        print("GPU Device:", torch.cuda.get_device_name(0))
-        print("GPU Memory:", torch.cuda.get_device_properties(0).total_memory / 1024**3, "GB")
-    
-    try:
-        nvidia_smi = subprocess.check_output(["nvidia-smi"])
-        print("nvidia-smi output:")
-        print(nvidia_smi.decode())
-    except:
-        print("nvidia-smi not available")
-
-# Run GPU check at startup
-print("=== GPU Status Check ===")
-check_gpu_status()
-print("======================")
-
-# Rest of your imports
 import gradio as gr
 import easyocr
+import torch
 from transformers import pipeline, DistilBertTokenizer, DistilBertForSequenceClassification
 import numpy as np
 from PIL import Image
@@ -40,21 +14,8 @@ if torch.cuda.is_available():
 else:
     print("Running on CPU")
 
-# Set device
-device = "cuda" if torch.cuda.is_available() else "cpu"
-print(f"Using device: {device}")
-
-# Initialize OCR reader with device specification
-try:
-    reader = easyocr.Reader(['en'], gpu=torch.cuda.is_available())
-    print("EasyOCR initialized successfully")
-except Exception as e:
-    print(f"Error initializing EasyOCR: {str(e)}")
-    reader = easyocr.Reader(['en'], gpu=False)
-    print("Falling back to CPU for EasyOCR")
-
 # Initialize OCR reader
-reader = easyocr.Reader(['en'])
+reader = easyocr.Reader(['en'], gpu=torch.cuda.is_available())
 
 # Initialize compliance rules
 compliance_rules = ComplianceRules()
@@ -164,31 +125,25 @@ def analyze_ad_copy(image):
     
     return report_text
 
-# Create Gradio interface
-iface = gr.Interface(
+# Create Gradio interface with updated parameters
+demo = gr.Interface(
     fn=analyze_ad_copy,
-    inputs=[
-        gr.Image(
-            type="pil",
-            label="Upload Marketing Material",
-            height=300,
-            width=400,
-            image_mode="RGB",
-            scale=1,
-            source="upload",
-            tool="select"
-        )
-    ],
-    outputs=gr.Textbox(label="Compliance Report", lines=10),
+    inputs=gr.Image(
+        type="numpy",
+        label="Upload Marketing Material",
+        height=300,
+        width=400,
+    ),
+    outputs=gr.Textbox(
+        label="Compliance Report",
+        lines=10,
+        max_lines=20
+    ),
     title="Marketing Campaign Compliance Checker",
     description="Upload marketing material to check compliance with US (SEC), UK (FCA), and EU financial regulations.",
-    examples=[],
-    theme=gr.themes.Base(),
-    allow_flagging="never"
+    theme=gr.themes.Default()
 )
 
-# Launch the app with additional error handling
-try:
-    iface.launch(debug=True)
-except Exception as e:
-    print(f"Error launching interface: {str(e)}")
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -1,4 +1,4 @@
-torch
+torch>=2.0.0
 torchvision
 gradio>=4.8.0
 easyocr