Update app.py

app.py

@@ -31,7 +31,10 @@ def caption_image(model_choice, image_input, url_input, load_in_8bit):
         captioner = pipeline(task="image-to-text",
                             model=CAPTION_MODELS[model_choice],
                             max_new_tokens=30,
-                            device_map="cpu", model_kwargs=model_kwargs, use_fast=True
+                            device=0, # Set the device as CPU
+                            model_kwargs=model_kwargs, 
+                            torch_dtype=torch.float16,  # Set the floating point to Float16
+                            use_fast=True
                             )
         # Store the loaded model
         loaded_models[model_key] = captioner