Update app.py

app.py

@@ -10,7 +10,7 @@ import PIL.Image
 import torch
 from transformers import InstructBlipProcessor, InstructBlipForConditionalGeneration
 
-DESCRIPTION = "# [BLIP-2 VQA Ad Listing Analysis](https://github.com/salesforce/LAVIS/tree/main/projects/blip2)"
+DESCRIPTION = "# [BLIP-2 test](https://github.com/salesforce/LAVIS/tree/main/projects/blip2)"
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU.</p>"
@@ -36,17 +36,7 @@ def answer_ad_listing_question(
     """
 
     inputs = processor(images=image, text=prompt, return_tensors="pt").to(device, torch.float16)
-    generated_ids = model.generate(
-        **inputs,
-        do_sample=True,  # Nucleus sampling is applied
-        temperature=1.0,  # Default temperature
-        length_penalty=1.0,  # Default length penalty
-        repetition_penalty=1.5,  # Default repetition penalty
-        max_length=50,  # Default max length
-        min_length=1,  # Default min length
-        num_beams=5,  # Default number of beams
-        top_p=0.9,  # Default top_p for nucleus sampling
-    )
+    generated_ids = model.generate(**inputs)
     result = processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
     return result
 
@@ -61,14 +51,14 @@ with gr.Blocks() as demo:
     with gr.Group():
         # Image and ad title input
         image = gr.Image(type="pil")
-        ad_title = gr.Textbox(label="Ad Title", placeholder="Enter the ad title here", lines=1)
+        ad_title = gr.Textbox(label="Advertisement Title", placeholder="Enter the title here", lines=1)
 
         # Output section
-        answer_output = gr.Textbox(label="Ad Listing Analysis", show_label=True, placeholder="Response will appear here.")
+        answer_output = gr.Textbox(label="Analysis", show_label=True, placeholder="Response.")
 
         # Submit and clear buttons
         with gr.Row():
-            submit_button = gr.Button("Analyze Ad Listing", variant="primary")
+            submit_button = gr.Button("Analyze Listing", variant="primary")
             clear_button = gr.Button("Clear")
 
     # Logic to handle clicking on "Analyze Ad Listing"