Update app.py

app.py

@@ -20,7 +20,7 @@ def generate_caption_from_image(image_path):
 
 def generate_story_from_caption(caption):
     # Generate story based on caption
-    llm = HuggingFaceModel(api_token=os.getenv('HUGGING_FACE_API_TOKEN'),
+    llm = HuggingFaceHub(huggingfacehub_api_token=os.getenv('HUGGING_FACE_API_TOKEN'),
                             repo_id="tiiuae/falcon-7b-instruct",
                             verbose=False,
                             model_kwargs={"temperature": 0.2, "max_new_tokens": 4000})