Update hugging_face.yaml

hugging_face.yaml

@@ -1,45 +1,40 @@
 _id: hugging_face
-title: Hugging Face
+author: Anton Breslavskii | https://github.com/breslavsky
 description: ""
-version: 1
 readme: ""
+title: Hugging Face
 url: https://huggingface.co/PiperMy/Node-Packages/resolve/main/hugging_face.yaml
-author: Anton Breslavskii | https://github.com/breslavsky
+version: 2
 nodes:
   pulid_flux:
     _id: pulid_flux
     arrange:
-      x: 140
-      y: 60
+      x: 180
+      y: 260
     category:
+      _id: deep_swap
       id: deep_swap
-      title: en=Deep swap;ru=Глубокая замена
+      title: en=Magic portrait;ru=Магический портрет
     environment:
       HF_TOKEN:
         title: Hugging Face Token
         type: string
         scope: global
     inputs:
-      prompt:
+      person:
         order: 1
+        title: en=Person;ru=Персона
+        type: image
+      prompt:
+        order: 2
         title: en=Prompt;ru=Подсказка
         type: string
         required: true
-        default: portrait, color, cinematic
-      person:
-        title: en=Person;ru=Фото человека
-        type: image
-        required: true
-      imageSize:
-        title: Image size
-        type: string
-        default: 1024x1024
-        enum:
-          - 512x512
-          - 1024x1024
+        multiline: true
+        default: a portrait of a zombie
     outputs:
-      image:
-        title: ""
+      portrait:
+        title: Portrait
         type: image
     package: hugging_face
     script: |-
@@ -48,132 +43,47 @@ nodes:
           const { FatalError, NextNode } = DEFINITIONS;
 
           const HF_TOKEN = env?.variables?.get('HF_TOKEN');
-          if(!HF_TOKEN) {
+          if (!HF_TOKEN) {
               throw new FatalError('Please, set your API key Hugging Face');
           }
 
           const { Client } = await import("@gradio/client/dist/index.js");
-          const { person, prompt, imageSize } = inputs;
-          const [width, height] = (() => {
-              const [width, height] = imageSize.split('x');
-              return [parseInt(width), parseInt(height)];
-          })();
-          
+
+          const { person, prompt } = inputs;
           const { data: id_image } = await download(person);
-          const client = await Client.connect("PiperMy/PuLID-FLUX", {
+          const client = await Client.connect("PiperMy/tight-inversion-pulid-demo", {
               hf_token: HF_TOKEN
           });
 
           console.log('Send request');
           try {
               const { data } = await client.predict("/generate_image", {
-                  prompt,
+                  prompt: prompt || 'a portrait of a zombie',
                   id_image,
-                  start_step: 0,
-                  guidance: 2,
-                  seed: -1,
-                  true_cfg: 1,
-                  width,
-                  height,
-                  num_steps: 25,
-                  id_weight: 1,
-                  neg_prompt: "artefacts, bad face",
+                  id_weight: 0.5,
+                  num_steps: 16,
+                  guidance: 3.5,
+                  width: 1024,
+                  height: 1024,
+                  neg_prompt: "bad image",
+                  true_cfg: 5,
                   timestep_to_start_cfg: 0,
                   max_sequence_length: 128,
+                  start_step: 0,
+                  seed: -1,
+                  gamma: .5,
+                  eta: 0.7,
+                  s: 0,
+                  tau: 2,
               });
               const [{ url }] = data;
-              const { data: image } = await download(url);
-              return NextNode.from({ outputs: { image } });
+              const { data: portrait } = await download(url);
+              return NextNode.from({ outputs: { portrait } });
           } catch (e) {
               throw new FatalError(e.message);
           }
 
       }
     source: catalog
-    title: PuLID Flux
-    version: 1
-  deepseek_r1_8b_six:
-    _id: deepseek_r1_8b_six
-    arrange:
-      x: 210
-      y: 310
-    category:
-      id: llm_agents
-      title: en=Language Agents;ru=Языковые агенты
-    environment:
-      HF_TOKEN:
-        title: Hugging Face Token
-        type: string
-        scope: pipeline
-    inputs:
-      question:
-        order: 1
-        title: en=Question;ru=Вопрос
-        type: string
-        required: true
-        multiline: true
-        default: What time is it now? Only JSON ready to parse.
-      answerFormat:
-        order: 2
-        title: en=Answer format;ru=Формат ответа
-        description: Don't forget add instructions for LLM
-        type: string
-        required: true
-        default: text
-        enum:
-          - text|Text
-          - json|JSON
-    outputs:
-      answer:
-        title: en=Answer;ru=Ответ
-        type: string
-      json:
-        title: JSON
-        type: json
-    package: hugging_face
-    script: |
-      export async function run({ inputs }) {
-
-          const { FatalError, NextNode } = DEFINITIONS;
-          const OpenAI = require('openai');
-
-          const HF_TOKEN = env?.variables?.get('HF_TOKEN');
-          if (!HF_TOKEN) {
-              throw new FatalError('Please, set your API key for HF');
-          }
-
-          const { question, model, answerFormat } = inputs;
-
-          const openai = new OpenAI({
-              baseURL: 'https://jq20v0lfcxycc1z3.us-east-1.aws.endpoints.huggingface.cloud/v1/',
-              apiKey: HF_TOKEN,
-          });
-
-          const result = await openai.chat.completions.create({
-              model: 'lmstudio-community/DeepSeek-R1-Distill-Llama-8B-GGUF',
-              messages: [
-                  { "role": "user", "content": question },
-              ],
-          });
-
-          const { content: answer } = result.choices[0].message;
-
-          switch (answerFormat) {
-              case 'text':
-                  return NextNode.from({ outputs: { answer } });
-              case 'json':
-                  try {
-                      const json = answer.replace(/^\`\`\`json\s*/ig, '').replace(/\`\`\`\s*$/ig, '');
-                      return NextNode.from({ outputs: { json: JSON.parse(json) } });
-                  } catch (e) {
-                      console.log(e);
-                      message(`Wrong JSON for question \`\`\`text\n${question}\n\`\`\`\nnanswer from LLM\n\`\`\`text${answer}\n\`\`\``, 'defect');
-                      throw new FatalError("Can't parse JSON asnwer from LLM");
-                  }
-              default:
-                  throw new FatalError(`Wrong answer format ${answerFormat}`);
-          }
-      }
-    source: catalog
-    title: DeepSeek R1 Distill Llama 8B Q8
+    title: Magic portrait
     version: 1