dipankardas011 commited on
Commit
ff29d15
·
unverified ·
1 Parent(s): 7852f50

added resoposen

Browse files

Signed-off-by: Dipankar Das <[email protected]>

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -31,6 +31,6 @@ def generate():
31
  text = "A picture of"
32
  inputs = processor(text=text, images=image, return_tensors="pt", add_special_tokens=False)
33
 
34
- generated_ids = model.generate(**inputs, max_new_tokens=50)
35
- generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
36
- print(generated_text)
 
31
  text = "A picture of"
32
  inputs = processor(text=text, images=image, return_tensors="pt", add_special_tokens=False)
33
 
34
+ generated_ids1 = model.generate(**inputs, max_new_tokens=50)
35
+ generated_text1 = processor.batch_decode(generated_ids1, skip_special_tokens=True)[0]
36
+ return {"output": {"autoregressive": generated_text, "conditional": generated_text1},}