Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -63,7 +63,7 @@ def stream_chat(message, history: list, system: str, temperature: float, max_new
|
|
63 |
image = None
|
64 |
|
65 |
prompt = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
|
66 |
-
inputs = processor(prompt, images=
|
67 |
|
68 |
generate_kwargs = dict(
|
69 |
max_new_tokens=max_new_tokens,
|
|
|
63 |
image = None
|
64 |
|
65 |
prompt = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
|
66 |
+
inputs = processor(prompt, images=image, return_tensors="pt").to(0)
|
67 |
|
68 |
generate_kwargs = dict(
|
69 |
max_new_tokens=max_new_tokens,
|