Update app.py

app.py

@@ -56,7 +56,8 @@ config = GeckoConfig.from_pretrained(model,
                                     print_keyword=print_keyword)
 processor = GeckoProcessor.from_pretrained(model, config=config, use_keyword=True, cropping_method=cropping_method, crop_size=crop_size)
 model = GeckoForConditionalGeneration.from_pretrained(
-    model, config=config)
+    model, config=config, torch_dtype=torch_dtype, 
+    attn_implementation=attn_implementation, device_map=device_map)
 model.load_text_encoder(processor)
 
 @spaces.GPU