spaces fix

app.py

@@ -10,7 +10,7 @@ with gr.Blocks(theme=theme) as demo:
         gr.HTML('<img src="file/logos/COP-GEN-logo.png" style="max-width: 300px;">')
         gr.Markdown("# 🔵 COP-GEN-Beta: Unified Generative Modelling of COPernicus Imagery Thumbnails")
         gr.Markdown("### Miguel Espinosa, Valerio Marsocci, Yuru Jia, Elliot J. Crowley, Mikolaj Czerkawski")
-        gr.Markdown('[[Website](https://miquel-espinosa.github.io/cop-gen-beta/)] [[GitHub](https://github.com/miquel-espinosa/COP-GEN-Beta)] [[Model](https://huggingface.co/mespinosami/COP-GEN-Beta)] [[Dataset](https://huggingface.co/Major-TOM)]')
+        gr.Markdown('[[Website](https://miquel-espinosa.github.io/cop-gen-beta/)] [GitHub](https://github.com/miquel-espinosa/COP-GEN-Beta)] [[Model](https://huggingface.co/mespinosami/COP-GEN-Beta) [[Dataset](https://huggingface.co/Major-TOM)]')
         gr.Markdown('> ## ⚠️ NOTE: This is a prototype Beta model of COP-GEN. It is based on image thumbnails of [[Major TOM](https://huggingface.co/Major-TOM)] and does not yet support raw source data. The hillshade visualisation is used for elevation. The full model COP-GEN is coming soon.')
     
     with gr.Column(elem_classes="Main app"):

src/backend.py

@@ -202,6 +202,7 @@ def run_inference(config, nnet, autoencoder, img_tensors):
 
     return generated_samples
 
+@spaces.GPU
 def custom_inference(images, generate_modalities, condition_modalities, num_inference_steps, seed=None):
     """
     Run custom inference with user-specified parameters

src/utils.py

@@ -160,8 +160,4 @@ def sample_shuffle():
     """
     grid_cell = grid_cell_df.sample().iloc[0]
 
-    return resize_and_crop(get_images(grid_cell))
-
-@spaces.GPU
-def generate_output():
-    pass
+    return resize_and_crop(get_images(grid_cell))