diff --git a/WebUI/src/App.vue b/WebUI/src/App.vue
index 8cce1c57..7a5c7087 100644
--- a/WebUI/src/App.vue
+++ b/WebUI/src/App.vue
@@ -66,17 +66,19 @@
         @show-download-model-confirm="showDownloadModelConfirm"></create>
       <enhance v-show="activeTabIdx == 1" ref="enhanceCompt" @show-download-model-confirm="showDownloadModelConfirm">
       </enhance>
-      <answer v-show="activeTabIdx == 2" @show-download-model-confirm="showDownloadModelConfirm"></answer>
+      <answer v-show="activeTabIdx == 2" ref = "answer" @show-download-model-confirm="showDownloadModelConfirm" @show-model-request="showModelRequest"></answer>
       <learn-more v-show="activeTabIdx == 3"></learn-more>
       <app-settings v-if="showSetting" @close="hideAppSettings" @show-download-model-confirm="showDownloadModelConfirm"></app-settings>
     </div>
     <download-dialog v-show="showDowloadDlg" ref="downloadDigCompt" @close="showDowloadDlg = false"></download-dialog>
+    <add-l-l-m-dialog v-show="showModelRequestDialog" ref="addLLMCompt" @close="showModelRequestDialog = false" @call-check-model="callCheckModel" @show-warning="showWarning"></add-l-l-m-dialog>
+    <warning-dialog v-show="showWarningDialog" ref="warningCompt" @close="showWarningDialog = false"></warning-dialog>
   </main>
   <footer class="flex-none px-4 flex justify-between items-center select-none" :class="{'bg-black bg-opacity-50': theme.active === 'lnl', 'bg-black bg-opacity-80': theme.active === 'bmg', 'border-t border-color-spilter': theme.active === 'dark'}">
     <div>
       <p>Al Playground from Intel Corporation <a href="https://github.com/intel/ai-playground" target="_blank"
           class="text-blue-500">https://github.com/intel/ai-playground</a></p>
-      <p>AI Playground version: v{{ productVersion }} 
+      <p>AI Playground version: v{{ productVersion }}
         <a href="https://github.com/intel/ai-playground/blob/main/AI%20Playground%20Users%20Guide.pdf" target="_blank"
         class="text-blue-500">  User Guide</a>
 
@@ -115,6 +117,8 @@ import { useGlobalSetup } from "./assets/js/store/globalSetup";
 import DownloadDialog from '@/components/DownloadDialog.vue';
 import { Collapsible, CollapsibleContent, CollapsibleTrigger } from '@/components/ui/collapsible'
 import { useTheme } from "./assets/js/store/theme.ts";
+import AddLLMDialog from "@/components/AddLLMDialog.vue";
+import WarningDialog from "@/components/WarningDialog.vue";
 
 
 const isOpen = ref(false);
@@ -127,12 +131,22 @@ const showSetting = ref(false);
 
 const enhanceCompt = ref<InstanceType<typeof Enhance>>();
 
+const answer = ref<InstanceType<typeof Answer>>();
+
 const showSettingBtn = ref<HTMLButtonElement>();
 
 const showDowloadDlg = ref(false);
 
+const showModelRequestDialog = ref(false);
+
+const showWarningDialog = ref(false);
+
 const downloadDigCompt = ref<InstanceType<typeof DownloadDialog>>();
 
+const addLLMCompt = ref<InstanceType<typeof AddLLMDialog>>();
+
+const warningCompt = ref<InstanceType<typeof WarningDialog>>();
+
 const fullscreen = ref(false);
 
 const platformTitle = window.envVars.platformTitle;
@@ -161,7 +175,6 @@ onBeforeMount(async () => {
   })
 })
 
-
 function showAppSettings() {
   if (showSetting.value === false) {
     showSetting.value = true;
@@ -223,4 +236,25 @@ function showDownloadModelConfirm(downList: DownloadModelParam[], success?: () =
     downloadDigCompt.value!.showConfirm(downList, success, fail);
   });
 }
+
+function showModelRequest() {
+  showModelRequestDialog.value = true;
+  nextTick(() => {
+    addLLMCompt.value!.onShow();
+  });
+}
+
+function callCheckModel(){
+  answer.value!.checkModel();
+}
+
+function showWarning(message : string, func : () => void) {
+  warningCompt.value!.warningMessage = message;
+  showWarningDialog.value = true;
+  warningCompt.value!.confirmFunction = func;
+  nextTick(() => {
+    warningCompt.value!.onShow();
+  });
+}
+
 </script>
diff --git a/WebUI/src/assets/i18n/en-US.json b/WebUI/src/assets/i18n/en-US.json
index 06315107..892d8ab3 100644
--- a/WebUI/src/assets/i18n/en-US.json
+++ b/WebUI/src/assets/i18n/en-US.json
@@ -3,9 +3,11 @@
   "COM_SETTINGS": "Settings",
   "COM_MINI": "Minimize",
   "COM_CLOSE": "Close",
+  "COM_ADD": "Add",
   "COM_RESTORE": "Restore Defaults",
   "COM_SD_PROMPT": "Input prompt to generate image",
   "COM_LLM_PROMPT": "Input prompt to generate answer",
+  "COM_LLM_HF_PROMPT": "<namespace>/<repo_name>",
   "COM_CLICK_UPLOAD": "- Click To Upload Image -",
   "COM_GENERATE": "Generate",
   "COM_GENERATING": "Generating",
@@ -123,6 +125,7 @@
   "DECREASE_FONT_SIZE": "Shrink Text",
   "ANSWER_RAG_ENABLE":"Enable File Query",
   "ANSWER_RAG_OPEN_DIALOG":"Open File Uploader",
+  "REQUEST_LLM_MODEL_NAME":"Add a model of your choice from huggingface.co,<br />for example: <i>meta-llama/Llama-3.2-1B </i>",
   "DOWNLOADER_CONFRIM_TIP":"You are missing one or more models needed to run. Would you like to download the model(s) listed below?",
   "DOWNLOADER_MODEL":"Model",
   "DOWNLOADER_INFO":"Info",
@@ -159,10 +162,13 @@
   "ERROR_RUNTIME_ERROR": "If a critical failure occurs, please restart the program and try again",
   "ERROR_GENERATE_UNKONW_EXCEPTION": "An unknown error occurred. failed to generate from model",
   "ERROR_FOLDER_NOT_EXISTS": "The specified directory does not exist",
+  "ERROR_REPO_NOT_EXISTS": "The specified repository-ID could not be found",
+  "ERROR_ALREADY_IN_MODELS": "The specified model is already in the list",
   "ERROR_ENHANCE_IMAGE_NOT_SET": "Please configure the input image to be generated",
   "ERROR_UNFOUND_GRAPHICS": "If the necessary hardware for program execution is not detected on the computer device, the program will terminate upon clicking OK.",
   "ERROR_PYTHON_BACKEND_INIT": "Backend initialization failed",
   "ERROR_PYTHON_BACKEND_INIT_DETAILS_TEXT": "The AI inference backend failed to initialize. Please try restarting the application. If the problem persists, you can check the Details for additional information about the error.",
   "ERROR_PYTHON_BACKEND_INIT_DETAILS": "Details",
-  "ERROR_PYTHON_BACKEND_INIT_OPEN_LOG": "Open Log"
+  "ERROR_PYTHON_BACKEND_INIT_OPEN_LOG": "Open Log",
+  "WARNING_MODEL_TYPE_WRONG": "The model type doesn't seem to fit the requirements. Are you sure, you want to continue?"
 }
diff --git a/WebUI/src/assets/js/store/models.ts b/WebUI/src/assets/js/store/models.ts
index 7782acc1..4f96b650 100644
--- a/WebUI/src/assets/js/store/models.ts
+++ b/WebUI/src/assets/js/store/models.ts
@@ -1,6 +1,6 @@
 import { defineStore } from "pinia";
 
-type ModelType = "llm" | "embedding" | "stableDiffusion" | "inpaint" | "lora" | "vae";
+export type ModelType = "llm" | "embedding" | "stableDiffusion" | "inpaint" | "lora" | "vae" | "undefined";
 
 export type Model = {
     name: string;
@@ -17,6 +17,9 @@ const predefinedModels: Model[] = [
     // { name: 'THUDM/chatglm3-6b', type: 'llm', downloaded: false },
 ]    
 
+export const userModels: Model[] = [
+]
+
 export const useModels = defineStore("models", () => {
 
     const hfToken = ref<string | undefined>(undefined);
@@ -39,10 +42,10 @@ export const useModels = defineStore("models", () => {
             ...inpaintModels.map<Model>(name => ({ name, type: 'inpaint', downloaded: true })),
             ...embeddingModels.map<Model>(name => ({ name, type: 'embedding', downloaded: true })),
         ];
+
         const notYetDownloaded = (model: Model) => !downloadedModels.map(m => m.name).includes(model.name);
 
-        models.value = [...downloadedModels, ...predefinedModels.filter(notYetDownloaded)];
-        console.log(models);
+        models.value = [...downloadedModels, ...userModels, ...predefinedModels.filter(notYetDownloaded)];
 
         }
 
diff --git a/WebUI/src/components/AddLLMDialog.vue b/WebUI/src/components/AddLLMDialog.vue
new file mode 100644
index 00000000..a80d68d2
--- /dev/null
+++ b/WebUI/src/components/AddLLMDialog.vue
@@ -0,0 +1,106 @@
+<template>
+  <div class="dialog-container z-10">
+    <div class="dialog-mask absolute left-0 top-0 w-full h-full bg-black/55 flex justify-center items-center">
+      <div
+        class="py-10 px-20 w-500px flex flex-col items-center justify-center bg-gray-600 rounded-3xl gap-6 text-white"
+        :class="{ 'animate-scale-in': animate }">
+        <p v-html="i18nState.REQUEST_LLM_MODEL_NAME"></p>
+        <Input :placeholder="languages.COM_LLM_HF_PROMPT" v-model="modelRequest" @keyup.enter="addModel"></Input>
+        <p v-show="addModelError" style="color: #F44336;">{{ addModelErrorMessage }}</p>
+        <div class="flex justify-center items-center gap-9">
+          <button @click="closeAdd" class="bg-color-control-bg  py-1 px-4 rounded">{{ i18nState.COM_CLOSE }}</button>
+          <button @click="addModel" class="bg-color-control-bg  py-1 px-4 rounded">{{ i18nState.COM_ADD }}</button>
+        </div>
+      </div>
+    </div>
+  </div>
+</template>
+<script setup lang="ts">
+import { Input } from '@/components/ui/input'
+import { useGlobalSetup } from '@/assets/js/store/globalSetup';
+import { useI18N } from '@/assets/js/store/i18n';
+import { useModels, userModels } from '@/assets/js/store/models';
+
+
+const i18nState = useI18N().state;
+const globalSetup = useGlobalSetup();
+const models = useModels();
+const modelRequest = ref("");
+const addModelErrorMessage = ref("")
+const addModelError = ref(false);
+const animate = ref(false);
+const emits = defineEmits<{
+  (e: "close"): void,
+  (e: "callCheckModel"): void,
+  (e: "showWarning", warning: string, func: () => void): void
+}>();
+
+function onShow() {
+  animate.value = true;
+}
+
+async function addModel() {
+  const previousModel = globalSetup.modelSettings.llm_model
+  const isInModels = models.models.some((model) => model.name === modelRequest.value)
+
+  const cancelAndShowWarning = (text: string) => {
+    globalSetup.modelSettings.llm_model = previousModel;
+    addModelErrorMessage.value = text;
+    addModelError.value = true;
+  }
+
+  if (isInModels) {
+    cancelAndShowWarning(i18nState.ERROR_ALREADY_IN_MODELS);
+    return;
+  }
+
+  const urlExists = await checkIfUrlExists(modelRequest.value);
+  if (!urlExists) {
+    cancelAndShowWarning(i18nState.ERROR_REPO_NOT_EXISTS);
+    return;
+  }
+
+  addModelError.value = false;
+
+  const isLlm = await isLLM(modelRequest.value);
+  const downloadNewModel = async () => {
+    await registerModel();
+    emits("callCheckModel");
+    closeAdd();
+  };
+
+  if (!isLlm) {
+    emits("showWarning", i18nState.WARNING_MODEL_TYPE_WRONG, downloadNewModel);
+  } else {
+    downloadNewModel();
+  }
+}
+
+async function registerModel() {
+  userModels.push({ name: modelRequest.value, type: 'llm', downloaded: false })
+  await models.refreshModels()
+  globalSetup.modelSettings.llm_model = modelRequest.value;
+}
+
+async function checkIfUrlExists(repo_id: string) {
+  const response = await fetch(`${globalSetup.apiHost}/api/checkHFRepoExists?repo_id=${repo_id}`)
+  const data = await response.json()
+  return data.exists;
+}
+
+async function isLLM(repo_id: string) {
+  const response = await fetch(`${globalSetup.apiHost}/api/isLLM?repo_id=${repo_id}`)
+  const data = await response.json()
+  return data.isllm
+}
+
+function closeAdd() {
+  addModelErrorMessage.value = "";
+  addModelError.value = false;
+  modelRequest.value = "";
+  emits("close");
+}
+
+defineExpose({ onShow });
+
+</script>
\ No newline at end of file
diff --git a/WebUI/src/components/DownloadDialog.vue b/WebUI/src/components/DownloadDialog.vue
index 7488bcf5..81a9c3ce 100644
--- a/WebUI/src/components/DownloadDialog.vue
+++ b/WebUI/src/components/DownloadDialog.vue
@@ -111,10 +111,6 @@ const readTerms = ref(false);
 const downloadList = ref<DownloadModelRender[]>([]);
 
 
-onDeactivated(() => {
-    animate.value = false;
-})
-
 function dataProcess(line: string) {
     console.log(line);
     const dataJson = line.slice(5);
diff --git a/WebUI/src/components/InpaintMask.vue b/WebUI/src/components/InpaintMask.vue
index c660d3da..d8b07ec5 100644
--- a/WebUI/src/components/InpaintMask.vue
+++ b/WebUI/src/components/InpaintMask.vue
@@ -108,7 +108,6 @@ function moveWithShadow(e: MouseEvent) {
 
 function showShadow() {
     shadowPos.show = true;
-    console.log("showShadow");
 }
 
 defineExpose({ clearMaskImage, getMaskImage })
diff --git a/WebUI/src/components/WarningDialog.vue b/WebUI/src/components/WarningDialog.vue
new file mode 100644
index 00000000..7cfe91a9
--- /dev/null
+++ b/WebUI/src/components/WarningDialog.vue
@@ -0,0 +1,41 @@
+<template>
+    <div class="dialog-container z-10">
+        <div class="dialog-mask absolute left-0 top-0 w-full h-full bg-black/55 flex justify-center items-center">
+            <div class="py-10 px-20 w-500px flex flex-col items-center justify-center bg-gray-600 rounded-3xl gap-6 text-white"
+                :class="{ 'animate-scale-in': animate }">
+              <p v-html= "warningMessage"></p>
+              <div class="flex justify-center items-center gap-9">
+                <button @click="cancelConfirm" class="bg-color-control-bg  py-1 px-4 rounded">{{i18nState.COM_CANCEL}}</button>
+                <button @click="confirmAdd" class="bg-color-control-bg  py-1 px-4 rounded">{{i18nState.COM_CONFIRM}}</button>
+              </div>
+            </div>
+        </div>
+    </div>
+</template>
+<script setup lang="ts">
+import { useI18N } from '@/assets/js/store/i18n.ts';
+const i18nState = useI18N().state;
+const confirmFunction = ref(() => {})
+const warningMessage = ref("")
+const animate = ref(false);
+const emits = defineEmits<{
+    (e: "close"): void
+}>();
+
+
+async function confirmAdd() {
+  confirmFunction.value()
+  emits("close");
+}
+
+function cancelConfirm() {
+  emits("close");
+}
+
+function onShow(){
+  animate.value = true
+}
+
+defineExpose({warningMessage, confirmFunction, onShow });
+
+</script>
\ No newline at end of file
diff --git a/WebUI/src/components/ui/input/Input.vue b/WebUI/src/components/ui/input/Input.vue
index a7d9ed70..a16a1ef9 100644
--- a/WebUI/src/components/ui/input/Input.vue
+++ b/WebUI/src/components/ui/input/Input.vue
@@ -20,5 +20,5 @@ const modelValue = useVModel(props, 'modelValue', emits, {
 </script>
 
 <template>
-  <input type="password" v-model="modelValue" :class="cn('flex h-9 w-full rounded-md border bg-[var(--textbox-bg)] px-3 py-1 text-sm shadow-sm transition-colors file:border-0 file:bg-transparent file:text-sm file:font-medium placeholder:text-slate-500 focus-visible:outline-none focus-visible:border-[var(--color-active)] disabled:cursor-not-allowed disabled:opacity-50 ', props.class)">
+  <input v-model="modelValue" :class="cn('flex h-9 w-full rounded-md border bg-[var(--textbox-bg)] px-3 py-1 text-sm shadow-sm transition-colors file:border-0 file:bg-transparent file:text-sm file:font-medium placeholder:text-slate-500 focus-visible:outline-none focus-visible:border-[var(--color-active)] disabled:cursor-not-allowed disabled:opacity-50 ', props.class)">
 </template>
diff --git a/WebUI/src/views/Answer.vue b/WebUI/src/views/Answer.vue
index 89523f4b..56c60963 100644
--- a/WebUI/src/views/Answer.vue
+++ b/WebUI/src/views/Answer.vue
@@ -104,6 +104,7 @@
                                 <model-drop-down-item :model="slotItem.item"></model-drop-down-item>
                             </template>
                         </drop-selector>
+                        <button class="svg-icon i-generate-add w-10 h-10 text-purple-500 ml-1.5" @click="addLLMModel"></button>
                         <button class="svg-icon i-refresh w-5 h-5 text-purple-500 flex-none ml-1"
                             @animationend="removeRonate360" @click="refreshLLMModles"></button>
                         <!-- <button
@@ -175,9 +176,6 @@
             </div>
             <rag v-if="ragData.showUploader" ref="ragPanel" @close="ragData.showUploader = false"></rag>
         </div>
-        <teleport to="#answerPanel" v-if="showDowloadDlg">
-            <download-dialog ref="downloadDigCompt"></download-dialog>
-        </teleport>
     </div>
 
 </template>
@@ -186,7 +184,6 @@ import Rag from "../components/Rag.vue";
 import ProgressBar from "../components/ProgressBar.vue";
 import LoadingBar from "../components/LoadingBar.vue";
 import DropSelector from "@/components/DropSelector.vue";
-import DownloadDialog from "@/components/DownloadDialog.vue";
 import ModelDropDownItem from "@/components/ModelDropDownItem.vue";
 import { useI18N } from '@/assets/js/store/i18n';
 import { toast } from '@/assets/js/toast';
@@ -221,17 +218,18 @@ const loadingModel = ref(false);
 let receiveOut = "";
 let chatPanel: HTMLElement;
 const markdownParser = new MarkdownParser(i18nState.COM_COPY);
-const showDowloadDlg = ref(false);
 const ragData = reactive({
     enable: false,
     processEnable: false,
     showUploader: false,
 });
-const downloadDigCompt = ref<InstanceType<typeof DownloadDialog>>()
+
 const source = ref("");
 const emits = defineEmits<{
     (e: "showDownloadModelConfirm", downloadList: DownloadModelParam[], success?: () => void, fail?: () => void): void,
+    (e: "showModelRequest", success?: () => void, fail?: () => void): void
 }>();
+
 let abortContooler: AbortController | null;
 const stopping = ref(false);
 const fontSizeIndex = ref(1); // sets default to text-sm
@@ -390,14 +388,16 @@ async function simulatedInput() {
 }
 
 function fastGenerate(e: KeyboardEvent) {
-    if (e.code == "Enter") {
-        if (e.ctrlKey || e.shiftKey || e.altKey) {
-            question.value += "\n";
-        } else {
-            e.preventDefault();
-            newPromptGenerate();
-        }
+  if (e.code == "Enter") {
+    if (e.ctrlKey || e.shiftKey || e.altKey) {
+      question.value += "\n";
+    } else {
+      e.preventDefault();
+      if (question.value !== "") {
+        newPromptGenerate()
+      }
     }
+  }
 }
 
 async function newPromptGenerate() {
@@ -420,7 +420,7 @@ async function newPromptGenerate() {
 }
 
 async function checkModel() {
-    return new Promise<void>(async (resolve, reject) => {
+  return new Promise<void>(async (resolve, reject) => {
         const checkList: CheckModelExistParam[] = [{ repo_id: globalSetup.modelSettings.llm_model, type: Const.MODEL_TYPE_LLM }];
         if (!(await globalSetup.checkModelExists(checkList))[0].exist) {
             emits(
@@ -435,6 +435,8 @@ async function checkModel() {
     });
 }
 
+
+
 async function generate(chatContext: ChatItem[]) {
     if (processing.value || chatContext.length == 0) { return; }
 
@@ -500,6 +502,13 @@ function changeLLMModel(model: Model, _: number) {
     globalSetup.applyModelSettings({ llm_model: model.name });
 }
 
+async function addLLMModel() {
+    return new Promise<void>(async (resolve, reject) => {
+      emits("showModelRequest", resolve, reject);
+    })
+
+}
+
 async function refreshLLMModles(e: Event) {
     const button = e.target as HTMLElement;
     button.classList.add("animate-ronate360");
@@ -574,5 +583,8 @@ async function enableRag() {
     ragData.processEnable = false;
 }
 
+defineExpose({
+  checkModel
+})
 
 </script>
\ No newline at end of file
diff --git a/WebUI/src/views/AppSettings.vue b/WebUI/src/views/AppSettings.vue
index 7a389cec..54b3d98a 100644
--- a/WebUI/src/views/AppSettings.vue
+++ b/WebUI/src/views/AppSettings.vue
@@ -36,7 +36,7 @@
                 <div class="flex flex-col gap-3">
                     <p>{{ languages.SETTINGS_MODEL_HUGGINGFACE_API_TOKEN }}</p>
                     <div class="flex flex-col items-start gap-1">
-                        <Input v-model="models.hfToken" :class="{ 'border-red-500': models.hfToken && !models.hfTokenIsValid }"/>
+                        <Input type="password" v-model="models.hfToken" :class="{ 'border-red-500': models.hfToken && !models.hfTokenIsValid }"/>
                         <div class="text-xs text-red-500 select-none" :class="{'opacity-0': !(models.hfToken && !models.hfTokenIsValid)}">{{ languages.SETTINGS_MODEL_HUGGINGFACE_INVALID_TOKEN_TEXT }}</div>
                     </div>
                 </div>
diff --git a/service/model_downloader.py b/service/model_downloader.py
index 538da864..a9df3b9d 100644
--- a/service/model_downloader.py
+++ b/service/model_downloader.py
@@ -1,4 +1,5 @@
 from huggingface_hub import HfFileSystem, hf_hub_url, model_info
+from huggingface_hub.utils import RepositoryNotFoundError
 from typing import Any, Callable, Dict, List
 from os import path, makedirs, rename
 import requests
@@ -11,11 +12,11 @@
 import traceback
 import concurrent.futures
 import aipg_utils as utils
+import shutil
 
 model_list_cache = dict()
 model_lock = Lock()
 
-
 class HFFileItem:
     relpath: str
     size: int
@@ -83,6 +84,16 @@ def __init__(self, hf_token=None) -> None:
         self.thread_lock = Lock()
         self.hf_token = hf_token
 
+    def hf_url_exists(self, repo_id: str):
+        try:
+            model_info(repo_id)
+            return True
+        except RepositoryNotFoundError:
+            return False
+
+    def probe_type(self, repo_id : str):
+        return model_info(repo_id).pipeline_tag
+
     def is_gated(self, repo_id: str):
         try:
             info = model_info(repo_id)
@@ -258,6 +269,9 @@ def multiple_thread_downlod(self, thread_count: int):
                     path.join(self.save_path, self.repo_id.replace("/", "---"))
                 ),
             )
+        else:
+            # Download aborted
+            shutil.rmtree(self.save_path_tmp)
 
     def start_report_download_progress(self):
         thread = Thread(target=self.report_download_progress)
diff --git a/service/web_api.py b/service/web_api.py
index 3ac03ce9..5cf6fa69 100644
--- a/service/web_api.py
+++ b/service/web_api.py
@@ -196,6 +196,33 @@ def check_model_exist():
     return jsonify({"code": 0, "message": "success", "exists": result_list})
 
 
+@app.route("/api/checkHFRepoExists", methods=["GET"])
+def check_if_huggingface_repo_exists():
+    repo_id = request.args.get('repo_id')
+    downloader = HFPlaygroundDownloader()
+    exists = downloader.hf_url_exists(repo_id)
+
+    return jsonify(
+            {
+                "exists": exists
+            }
+        )
+
+@app.route("/api/isLLM", methods=["GET"])
+def is_llm():
+    repo_id = request.args.get('repo_id')
+    downloader = HFPlaygroundDownloader()
+    try:
+        model_type_hf = downloader.probe_type(repo_id)
+    except Exception:
+        model_type_hf = "undefined"
+    return jsonify(
+            {
+            "isllm": model_type_hf == "text-generation"
+            }
+        )
+
+
 size_cache = dict()
 lock = threading.Lock()