Spaces:

mrm8488
/

idefics-9b-ft-describe-diffusion-mj

Runtime error

multimodalart HF staff commited on Sep 26, 2023

Commit

a853a81

•

1 Parent(s): 5bbc196

Remove Space (#3)

- Remove Space (e5113d08dced01513564c456ec99530c71b48381)

Co-authored-by: Apolinário from multimodal AI art <multimodalart@users.noreply.huggingface.co>

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ def predict(image):
     inputs = processor(prompts[0], return_tensors="pt").to(device)
     generated_ids = model.generate(**inputs, max_length=max_length)
     generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
-    generated_text = generated_text.replace(prompt,"")
     return generated_text
 title = "Midjourney-like Image Captioning with IDEFICS"

     inputs = processor(prompts[0], return_tensors="pt").to(device)
     generated_ids = model.generate(**inputs, max_length=max_length)
     generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
+    generated_text = generated_text.replace(f"{prompt} ","")
     return generated_text
 title = "Midjourney-like Image Captioning with IDEFICS"