Update model urls

Author: a-szymanska

apps/computer-vision/app/text_to_image/index.tsx

@@ -49,11 +49,13 @@ export default function TextToImageScreen() {
 
   const runForward = async (input: string, numSteps: number) => {
     if (!input || !input.trim()) return;
+    const prevImageTitle = imageTitle;
     setImageTitle(input);
     setSteps(numSteps);
     try {
       const output = await model.generate(input, imageSize, steps);
       if (!output.length) {
+        setImageTitle(prevImageTitle);
         return;
       }
       setImage(output);
@@ -66,6 +68,7 @@ export default function TextToImageScreen() {
   };
 
   if (!model.isReady) {
+    // TODO: Update when #614 merged
     return (
       <Spinner
         visible={!model.isReady}

packages/react-native-executorch/src/constants/modelUrls.ts

@@ -425,27 +425,17 @@ export const CLIP_VIT_BASE_PATCH32_TEXT = {
 
 // Image generation
 export const BK_SDM_TINY_VPRED_512 = {
-  schedulerSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/scheduler/scheduler_config.json',
-  tokenizerSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/tokenizer/tokenizer.json',
-  encoderSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/text_encoder/model.pte',
-  unetSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/unet/model.pte',
-  decoderSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/vae/model.pte',
+  schedulerSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/scheduler/scheduler_config.json`,
+  tokenizerSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/tokenizer/tokenizer.json`,
+  encoderSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/text_encoder/model.pte`,
+  unetSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/unet/model.pte`,
+  decoderSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/vae/model.pte`,
 };
 
 export const BK_SDM_TINY_VPRED_256 = {
-  schedulerSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/scheduler/scheduler_config.json',
-  tokenizerSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/tokenizer/tokenizer.json',
-  encoderSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/text_encoder/model.pte',
-  unetSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/unet/model.256.pte',
-  decoderSource:
-    'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/vae/model.256.pte',
+  schedulerSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/scheduler/scheduler_config.json`,
+  tokenizerSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/tokenizer/tokenizer.json`,
+  encoderSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/text_encoder/model.pte`,
+  unetSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/unet/model.256.pte`,
+  decoderSource: `${URL_PREFIX}-bk-sdm-tiny/${VERSION_TAG}/vae/model.256.pte`,
 };