update app.py

.gitignore

@@ -0,0 +1 @@
+.gradio/certificate.pem

app.py

@@ -25,14 +25,22 @@ prompt = (
 
 def caption_image(image):
     inputs = processor(images=image, text=prompt, return_tensors="pt").to(device)
-    out = model.generate(**inputs, max_new_tokens=250, temperature=0.7, top_p=0.9)
-    return processor.batch_decode(out, skip_special_tokens=True)[0]
+    out = model.generate(
+        **inputs,
+        max_new_tokens=250,
+        do_sample=True,
+        temperature=0.7,
+        top_p=0.9,
+        repetition_penalty=1.2)
+    output = processor.batch_decode(out, skip_special_tokens=True)[0]
+    print(output)
+    return output
 
 demo = gr.Interface(
     fn=caption_image, 
-    inputs=gr.Image(type="pil"), 
+    inputs=gr.Image(type="pil", label="Upload an Image", height=400), 
     outputs="text",
     title="Image to Paragraph Captioning"
 )
 
-demo.launch()
+demo.queue(api_open=False, max_size=10).launch()