Update app.py

app.py

@@ -1,4 +1,14 @@
-from transformers import DetrImageProcessor, DetrForObjectDetection, pipeline
+import os
+import sys
+from subprocess import check_call
+
+# Ensure transformers is installed
+try:
+    from transformers import DetrImageProcessor, DetrForObjectDetection, pipeline
+except ImportError:
+    check_call([sys.executable, "-m", "pip", "install", "transformers==4.33.2"])
+    from transformers import DetrImageProcessor, DetrForObjectDetection, pipeline
+
 from PIL import Image
 import gradio as gr
 
@@ -7,7 +17,7 @@ detection_model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-5
 detection_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
 description_generator = pipeline("text-generation", model="gpt-2")
 
-# Function to recognize and describe product
+# Function to recognize and describe a product
 def recognize_and_describe(image):
     inputs = detection_processor(images=image, return_tensors="pt")
     outputs = detection_model(**inputs)