cmd gen
Browse files
src/lib/components/PicletGenerator/PicletGenerator.svelte
CHANGED
|
@@ -67,9 +67,21 @@
|
|
| 67 |
|
| 68 |
console.log(`Generating text with Command client...`);
|
| 69 |
|
| 70 |
-
//
|
| 71 |
-
|
| 72 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
};
|
| 74 |
|
| 75 |
let workflowState: PicletWorkflowState = $state({
|
|
|
|
| 67 |
|
| 68 |
console.log(`Generating text with Command client...`);
|
| 69 |
|
| 70 |
+
// Command client expects a message format with text and files
|
| 71 |
+
// Using the generate function (fn_index based on ChatInterface)
|
| 72 |
+
const message = {
|
| 73 |
+
text: prompt,
|
| 74 |
+
files: [] // No images for text-only generation
|
| 75 |
+
};
|
| 76 |
+
|
| 77 |
+
// ChatInterface expects: message, history, max_new_tokens
|
| 78 |
+
const result = await client.predict("/chat", {
|
| 79 |
+
message: message,
|
| 80 |
+
history: [], // Empty history for single-turn generation
|
| 81 |
+
max_new_tokens: 2000 // Max from the space
|
| 82 |
+
});
|
| 83 |
+
|
| 84 |
+
return result.data || '';
|
| 85 |
};
|
| 86 |
|
| 87 |
let workflowState: PicletWorkflowState = $state({
|