| const keywords = { | |
| image: [ | |
| "image", | |
| "draw", | |
| "paint", | |
| "picture", | |
| "photo", | |
| "art", | |
| "illustration", | |
| "generate image" | |
| ], | |
| translation: [ | |
| "translate", | |
| "translation", | |
| "convert language" | |
| ], | |
| summary: [ | |
| "summarize", | |
| "summary", | |
| "shorten" | |
| ], | |
| code: [ | |
| "code", | |
| "python", | |
| "java", | |
| "cpp", | |
| "c++", | |
| "program" | |
| ], | |
| speech: [ | |
| "speech", | |
| "voice", | |
| "audio" | |
| ], | |
| ocr: [ | |
| "ocr", | |
| "scan", | |
| "extract text" | |
| ], | |
| caption: [ | |
| "caption", | |
| "describe image" | |
| ] | |
| }; | |
| function detectModel(prompt){ | |
| prompt = prompt.toLowerCase(); | |
| for(let task in keywords){ | |
| for(let word of keywords[task]){ | |
| if(prompt.includes(word)){ | |
| return task; | |
| } | |
| } | |
| } | |
| return "text"; | |
| } | |
| document.getElementById("generateBtn").onclick=function(){ | |
| const prompt=document.getElementById("prompt").value; | |
| const task=detectModel(prompt); | |
| let model=""; | |
| let output=""; | |
| switch(task){ | |
| case "image": | |
| model="Stable Diffusion XL"; | |
| output="🖼 Image Generation Model Selected"; | |
| break; | |
| case "translation": | |
| model="NLLB-200"; | |
| output="🌍 Translation Model Selected"; | |
| break; | |
| case "summary": | |
| model="BART"; | |
| output="📄 Summarization Model Selected"; | |
| break; | |
| case "code": | |
| model="Code Llama"; | |
| output="💻 Code Generation Model Selected"; | |
| break; | |
| case "speech": | |
| model="Whisper"; | |
| output="🎤 Speech Recognition Model Selected"; | |
| break; | |
| case "ocr": | |
| model="TrOCR"; | |
| output="📷 OCR Model Selected"; | |
| break; | |
| case "caption": | |
| model="BLIP"; | |
| output="🖼 Image Captioning Model Selected"; | |
| break; | |
| default: | |
| model="Gemma 3"; | |
| output="💬 Text Generation Model Selected"; | |
| } | |
| document.getElementById("model").innerHTML=model; | |
| document.getElementById("output").innerHTML=output; | |
| } |