jamesoncrate commited on
Commit
185d799
·
1 Parent(s): 3eb8145

update filename

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -78,15 +78,15 @@ def generate_embeddings(prompts_text):
78
  prompt_embeds_dict = dict(zip(prompts, prompt_embeds_cpu))
79
 
80
  # Save to temporary file
81
- temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.pth')
82
- torch.save(prompt_embeds_dict, temp_file.name)
83
- temp_file.close()
84
 
85
  status_msg = f"✅ Successfully generated embeddings for {len(prompts)} prompts!\n"
86
  status_msg += "Each embedding has shape: [1, 77, 4096]\n"
87
  status_msg += "Prompts processed:\n" + "\n".join([f" - '{p}'" for p in prompts])
88
 
89
- return temp_file.name, status_msg
90
 
91
  except Exception as e:
92
  import traceback
 
78
  prompt_embeds_dict = dict(zip(prompts, prompt_embeds_cpu))
79
 
80
  # Save to temporary file
81
+ temp_dir = tempfile.gettempdir()
82
+ temp_file_path = os.path.join(temp_dir, 'prompt_embeds_dict.pth')
83
+ torch.save(prompt_embeds_dict, temp_file_path)
84
 
85
  status_msg = f"✅ Successfully generated embeddings for {len(prompts)} prompts!\n"
86
  status_msg += "Each embedding has shape: [1, 77, 4096]\n"
87
  status_msg += "Prompts processed:\n" + "\n".join([f" - '{p}'" for p in prompts])
88
 
89
+ return temp_file_path, status_msg
90
 
91
  except Exception as e:
92
  import traceback