Update app.py

app.py

@@ -14,13 +14,10 @@ model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-capt
 model_translater = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-en-ar")
 
 # conditional image captioning (with prefix-)
-def image_captioning(image_url, prefix="a "):
+def image_captioning(image, prefix="a "):
   """ Return text (As str) to describe an image """
-  # Get the image by image_url and convert it to RGB
-  raw_image = Image.open(requests.get(image_url, stream=True).raw).convert('RGB')
-
   # Process the image
-  inputs = processor(raw_image, prefix, return_tensors="pt")
+  inputs = processor(image, prefix, return_tensors="pt")
 
   # Generate text to describe the image
   output = model.generate(**inputs)
@@ -34,9 +31,9 @@ def translate_text(text, to="ar"):
   translated_text = model_translater(str(text))
   return translated_text[0]['translation_text']
 
-def image_captioning_ar(image_url, prefix = "a "):
-  if image_url:
-    text = image_captioning(image_url, prefix=prefix)
+def image_captioning_ar(image, prefix = "a "):
+  if image:
+    text = image_captioning(image, prefix=prefix)
     return translate_text(text)
   return null