Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -222,7 +222,7 @@ class UnifiedAISystem:
|
|
222 |
|
223 |
with self.pipe_lock:
|
224 |
self.internvl_pipe.start_chat()
|
225 |
-
output = self.internvl_pipe.generate(prompt, image=image_tensor, max_new_tokens=
|
226 |
self.internvl_pipe.finish_chat()
|
227 |
|
228 |
|
|
|
222 |
|
223 |
with self.pipe_lock:
|
224 |
self.internvl_pipe.start_chat()
|
225 |
+
output = self.internvl_pipe.generate(prompt, image=image_tensor, max_new_tokens=4090)
|
226 |
self.internvl_pipe.finish_chat()
|
227 |
|
228 |
|