Spaces:
Runtime error
Runtime error
Commit
·
29ffaa1
1
Parent(s):
4c7ba7b
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ def generate_answer_git(processor, model, image, question):
|
|
38 |
input_ids = [processor.tokenizer.cls_token_id] + input_ids
|
39 |
input_ids = torch.tensor(input_ids).unsqueeze(0)
|
40 |
|
41 |
-
generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=50)
|
42 |
generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
|
43 |
|
44 |
return generated_answer
|
@@ -48,7 +48,7 @@ def generate_answer_blip(processor, model, image, question):
|
|
48 |
# prepare image + question
|
49 |
inputs = processor(images=image, text=question, return_tensors="pt")
|
50 |
|
51 |
-
generated_ids = model.generate(**inputs, max_length=50)
|
52 |
generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
|
53 |
|
54 |
return generated_answer
|
|
|
38 |
input_ids = [processor.tokenizer.cls_token_id] + input_ids
|
39 |
input_ids = torch.tensor(input_ids).unsqueeze(0)
|
40 |
|
41 |
+
generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=256)#50)
|
42 |
generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
|
43 |
|
44 |
return generated_answer
|
|
|
48 |
# prepare image + question
|
49 |
inputs = processor(images=image, text=question, return_tensors="pt")
|
50 |
|
51 |
+
generated_ids = model.generate(**inputs, max_length=256)#50)
|
52 |
generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
|
53 |
|
54 |
return generated_answer
|