Update app.py

app.py

@@ -262,7 +262,7 @@ class Demo:
             loss.backward()
             optimizer.step()
 
-        torch.save(finetuner.state_dict(), 'ft.pt')
+        torch.save(finetuner.state_dict(), f'{prompt.lower().replace(' ', '')}.pt')
 
         self.finetuner = finetuner.eval().half()
 
@@ -272,9 +272,9 @@ class Demo:
 
         self.training = False      
 
-        model_map['Custom'] = 'ft.pt'      
+        model_map['Custom'] = f'{prompt.lower().replace(' ', '')}.pt' 
 
-        return [gr.update(interactive=True), gr.update(interactive=True), 'ft.pt', gr.Dropdown.update(choices=list(model_map.keys()), value='Custom')]
+        return [gr.update(interactive=True), gr.update(interactive=True), f'{prompt.lower().replace(' ', '')}.pt', gr.Dropdown.update(choices=list(model_map.keys()), value='Custom')]
 
 
     def inference(self, prompt, seed, pbar = gr.Progress(track_tqdm=True)):