Spaces:
Sleeping
Sleeping
update app
Browse files
app.py
CHANGED
|
@@ -292,8 +292,8 @@ def generate_image(model_name: str, text: str, image: Image.Image,
|
|
| 292 |
text=[prompt_full],
|
| 293 |
images=[image],
|
| 294 |
return_tensors="pt",
|
| 295 |
-
padding=True
|
| 296 |
-
truncation=False, # Disabled truncation
|
| 297 |
).to(device)
|
| 298 |
|
| 299 |
streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
|
|
|
|
| 292 |
text=[prompt_full],
|
| 293 |
images=[image],
|
| 294 |
return_tensors="pt",
|
| 295 |
+
padding=True
|
| 296 |
+
#truncation=False, # Disabled truncation
|
| 297 |
).to(device)
|
| 298 |
|
| 299 |
streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
|