Remove dowload button

app.py

@@ -25,12 +25,7 @@ class Demo:
 
     def disable(self):
         return [gr.update(interactive=False), gr.update(interactive=False)]
-    def save(self):
-
-        if self.finetuner is not None:
-
-            torch.save()
-
+  
     def layout(self):
 
         with gr.Row():
@@ -75,10 +70,6 @@ class Demo:
                 self.train_button = gr.Button(
                     value="Train",
                 )
-
-                self.download = gr.Button(value="Download Model Weights")
-                self.download.click(self.save)
-                
                 
             with gr.Column(scale=2) as inference_column: