Update app.py

app.py

@@ -26,12 +26,12 @@ torch.hub.download_url_to_file(
     "https://cdn.openai.com/dall-e-2/demos/text2im/astronaut/horse/photo/0.jpg", "astronaut.jpg"
 )
 
-@spaces.GPU(duration=30)
+# @spaces.GPU(duration=30)
 def run(image: Union[str, PIL.Image.Image]) -> str:
     if isinstance(image, str):
         image = Image.open(image)
     inputs = processor(images=image, return_tensors="pt").to(device)
-    out = model.generate(pixel_values=inputs.pixel_values, num_beams=3, max_length=300)
+    out = model.generate(pixel_values=inputs.pixel_values, num_beams=1, max_length=300)
     generated_caption = processor.decode(out[0], skip_special_tokens=True)
     return generated_caption