Update brain.py

brain.py

@@ -1,3 +1,4 @@
+import os
 from transformers import pipeline
 from langchain_huggingface import HuggingFaceEndpoint
 from langchain_core.prompts import PromptTemplate
@@ -42,9 +43,11 @@ class StoryGenerator:
 
     def generate_story_from_image(self, image, model_name):
         """Generate a story from an image."""
-        temp_image_path = "assets/temp_image.jpg"
+        temp_image_path = "assets/image.jpg"
+        image.save(temp_image_path)
         scenario = self.img2txt(temp_image_path)
         story = self.generate_story(scenario, model_name)
+        os.remove(temp_image_path)
         return story
 
 # Example usage