Spaces:
Runtime error
Runtime error
Commit
•
e9d7857
1
Parent(s):
f0e42ed
Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ resblock.load_state_dict(torch.load('./model_chkpt/step2_resblock.pth',map_locat
|
|
| 44 |
|
| 45 |
def model_generate_ans(img,val_q):
|
| 46 |
|
| 47 |
-
max_generate_length =
|
| 48 |
|
| 49 |
# image
|
| 50 |
image_processed = processor(images=img, return_tensors="pt").to(device)
|
|
|
|
| 44 |
|
| 45 |
def model_generate_ans(img,val_q):
|
| 46 |
|
| 47 |
+
max_generate_length = 30
|
| 48 |
|
| 49 |
# image
|
| 50 |
image_processed = processor(images=img, return_tensors="pt").to(device)
|