changed ft.pt to prompt-based

app.py

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