llava-next_Granite-Vision-3.1-2B

Running on Zero

sflindrs commited on Feb 11

Commit

6284e5a

verified ·

1 Parent(s): 98ac18a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,6 +15,7 @@ model.to("cuda:0")
 @spaces.GPU
 def bot_streaming(message, history):
   print(message)
   if message["files"]:
     image = message["files"][-1]["path"]
   else:

 @spaces.GPU
 def bot_streaming(message, history):
   print(message)
+  image = None
   if message["files"]:
     image = message["files"][-1]["path"]
   else: