Update app.py

app.py

@@ -17,18 +17,15 @@ model.eval()
 def predict(prompt, image_url, max_length):
     image = processor.image_processor.fetch_images(image_url)
     prompts = [[image, prompt]]
-
     inputs = processor(prompts[0], return_tensors="pt").to(device)
-
-    generated_ids = model.generate(**inputs, max_length=128)
+    generated_ids = model.generate(**inputs, max_length=max_length)
     generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
-    print(generated_text)
     return generated_text
 
 
 
 title = "Midjourney-like Image Captioning with IDEFICS"
-description = "Gradio Demo for generating Midjourney like captions (describe functionality) with IDEFICS"
+description = "Gradio Demo for generating *Midjourney* like captions (describe functionality) with **IDEFICS**"
 
 examples = [
     ["Describe the following image:", "https://cdn.arstechnica.net/wp-content/uploads/2023/06/zoomout_2-1440x807.jpg", 64],