RealVis_v5.0_BF16_IP_B

Running on Zero

1inkusFace commited on 21 days ago

Commit

4d5c246

verified ·

1 Parent(s): cb1fed0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -244,7 +244,7 @@ def captioning(img):
     inputsa = processor5(images=img, return_tensors="pt").to('cuda')
     generated_ids = model5.generate(**inputsa, min_length=24, max_length=42)
     generated_text = processor5.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
-    output_prompt.append.(generated_text)
     print(generated_text)
     # Loop through prompts array:
     for prompt in prompts_array:

     inputsa = processor5(images=img, return_tensors="pt").to('cuda')
     generated_ids = model5.generate(**inputsa, min_length=24, max_length=42)
     generated_text = processor5.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
+    output_prompt.append(generated_text)
     print(generated_text)
     # Loop through prompts array:
     for prompt in prompts_array: