Update app.py

app.py

@@ -11,10 +11,12 @@ story_tokenizer = AutoTokenizer.from_pretrained(story_model_name)
 story_model = AutoModelForCausalLM.from_pretrained(story_model_name)
 
 # Function to extract text description from an image
-def extract_description(image_path):
+def extract_description(image_array):
     try:
+        # Convert the NumPy array to a PIL image
+        image = Image.fromarray(image_array)
         # Use the OCR model to extract a caption/description from the image
-        result = ocr_model(Image.open(image_path))
+        result = ocr_model(image)
         return result[0]["generated_text"]
     except Exception as e:
         return f"Error extracting description: {e}"
@@ -39,7 +41,7 @@ def create_story(image):
         # Step 1: Extract a description from the image
         description = extract_description(image)
         if not description or "Error" in description:
-            return description, None
+            return description
 
         # Step 2: Generate a story from the extracted description
         story = generate_story(description)