Simba commited on
Commit
b4a3e90
·
1 Parent(s): 21e80f4
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -26,9 +26,13 @@ def save_image_to_drive(image: np.ndarray) -> str:
26
 
27
 
28
 
29
- def respond(image: np.ndarray, prompt: str, chat_history):
 
 
 
 
30
  image = np.fliplr(image)
31
- image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
32
  image_path = save_image_to_drive(image)
33
  response = connector.simple_prompt(image=image, prompt=prompt)
34
  chat_history.append(((image_path,), None))
 
26
 
27
 
28
 
29
+ def respond(image: np.ndarray, prompt: str, chat_history=None):
30
+ # Initialize chat_history as an empty list if it's None
31
+ if chat_history is None:
32
+ chat_history = []
33
+
34
  image = np.fliplr(image)
35
+ image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
36
  image_path = save_image_to_drive(image)
37
  response = connector.simple_prompt(image=image, prompt=prompt)
38
  chat_history.append(((image_path,), None))