Spaces:
Running
Running
Update apps/format.js
Browse files- apps/format.js +6 -3
apps/format.js
CHANGED
|
@@ -94,12 +94,15 @@ router.post('/generate', async (req, res) => {
|
|
| 94 |
|
| 95 |
if (type === 'image') {
|
| 96 |
// VISION ROUTE (e.g., Qwen 2 VL, Grok Vision, etc.)
|
| 97 |
-
modelToUse = "
|
| 98 |
-
|
|
|
|
|
|
|
| 99 |
userPrompt = "Analyze this chat screenshot and write my next reply.";
|
| 100 |
} else {
|
| 101 |
// TEXT ROUTE (e.g., DeepSeek, Evathene, Hermes, etc.)
|
| 102 |
-
modelToUse = "
|
|
|
|
| 103 |
userPrompt = `Write my reply to this message:\n\n"${content}"`;
|
| 104 |
}
|
| 105 |
|
|
|
|
| 94 |
|
| 95 |
if (type === 'image') {
|
| 96 |
// VISION ROUTE (e.g., Qwen 2 VL, Grok Vision, etc.)
|
| 97 |
+
modelToUse = "bytedance-1.6";
|
| 98 |
+
// modelToUse = "mistral-small";
|
| 99 |
+
|
| 100 |
+
imagesPayload = [content]; // base64 string
|
| 101 |
userPrompt = "Analyze this chat screenshot and write my next reply.";
|
| 102 |
} else {
|
| 103 |
// TEXT ROUTE (e.g., DeepSeek, Evathene, Hermes, etc.)
|
| 104 |
+
modelToUse = "bytedance-1.6";
|
| 105 |
+
// modelToUse = "mistral-small"; // Update this to match your ai_engine mapping
|
| 106 |
userPrompt = `Write my reply to this message:\n\n"${content}"`;
|
| 107 |
}
|
| 108 |
|