Update app.py

app.py

@@ -7,6 +7,10 @@ import torch
 from PIL import Image
 import argparse
 import spaces
+import os
+
+root_path = os.path.dirname(os.path.abspath(__file__))
+print(root_path)
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--model-path", type=str, default="TheFinAI/FinLLaVA")
@@ -58,6 +62,8 @@ def bot_streaming(message, history):
     # Use a streamer to generate the output in a streaming fashion
     streamer = []
 
+    image_file = image if isinstance(image, str) else image.filename
+
     # Define a function to call chat_llava in a separate thread
     def generate_output():
         output = chat_llava(