Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
185d799
1
Parent(s):
3eb8145
update filename
Browse files
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 |
-
|
| 82 |
-
|
| 83 |
-
|
| 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
|
| 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
|