Update app.py

app.py

@@ -17,54 +17,102 @@ logger = logging.getLogger(__name__)
 # Tambahkan 'hf_model_name' jika nama model di Hugging Face berbeda dari ID yang Anda inginkan.
 # Jika nama model di Hugging Face sama, tidak perlu 'hf_model_name'.
 model_info = {
-    "Albert-Base-V2": {"task": "fill-mask", "description": "BERT-based model for masked language modeling"},
-    "GPT-2": {"task": "text-generation", "description": "GPT-2 model for text generation"},
-    "Tinny-Llama": {"task": "text-generation", "description": "Lightweight LLaMA model"},
-    "Electra-Small": {"task": "fill-mask", "description": "Small ELECTRA model"},
-    "GPT-2-Tinny": {"task": "text-generation", "description": "Tiny GPT-2 variant"},
-    "Bert-Tinny": {"task": "fill-mask", "description": "Tiny BERT model"},
-    "Distilbert-Base-Uncased": {"task": "fill-mask", "description": "Distilled BERT model"},
-    "Pythia": {"task": "text-generation", "description": "Pythia language model"},
-    "T5-Small": {"task": "text2text-generation", "description": "Small T5 model", "hf_model_name": "t5-small"},
-    "GPT-Neo": {"task": "text-generation", "description": "GPT-Neo model"},
-    "Distil-GPT-2": {"task": "text-generation", "description": "Distilled GPT-2 model"},
-    # --- MODEL EXTERNAL ---
-    "TinyLLama-NSFW-Chatbot": {
+    "Albert-Base-V2": {
+        "task": "fill-mask", 
+        "description": "Lyon28 Albert-Base-V2"
+    },
+    "GPT-2": {
+        "task": "text-generation", 
+        "description": "Lyon28 GPT-2"
+    },
+    "Tinny-Llama": {
         "task": "text-generation",
-        "description": "BilalRahib's TinyLLama NSFW Chatbot",
-        "hf_model_name": "bilalRahib/TinyLLama-NSFW-Chatbot"
+        "description": "Lyon 28 Tinny Llama"
+    },
+    "Electra-Small": {
+        "task": "fill-mask", 
+        "description": "Lyon28 Small ELECTRA"
+    },
+    "GPT-2-Tinny": {
+        "task": "text-generation",
+        "description": "Lyon28 Tiny GPT-2"
+    },
+    "Bert-Tinny": {
+        "task": "fill-mask", 
+        "description": "Lyon28 Tiny BERT"
     },
-    "M-GPT": {
+    "Distilbert-Base-Uncased": {
+        "task": "fill-mask",
+        "description": "Lyon28 Distilled BERT"
+    },
+    "Pythia": {
         "task": "text-generation",
-        "description": "ai-forever mGPT",
-        "hf_model_name": "ai-forever/mGPT"
+        "description": "Lyon28 Pythia"
+    },
+    "T5-Small": {
+        "task": "text2text-generation", 
+        "description": "Lyon28 Small T5"
     },
-    "Devstral-Small-2505": {
+    "GPT-Neo": {
         "task": "text-generation",
-        "description": "mistralai Devstral-Small-2505",
-        "hf_model_name": "mistralai/Devstral-Small-2505"
+        "description": "Lyon28 GPT-Neo"
     },
-    "Llama-3_3-Nemotron-Super-49B-GenRM-Multilingual": {
+    "Distil-GPT-2": {
         "task": "text-generation",
-        "description": "nvidia Llama-3_3-Nemotron-Super-49B-GenRM-Multilingual",
-        "hf_model_name": "nvidia/Llama-3_3-Nemotron-Super-49B-GenRM-Multilingual"
+        "description": "Lyon28 Distilled GPT-2"
     },
-    "Dhanishtha-2.0-preview": {
+    # --- MODEL EXTERNAL ---
+    "TinyLLama-NSFW-Chatbot": {
         "task": "text-generation",
-        "description": "HelpingAI Dhanishtha-2.0-preview",
-        "hf_model_name": "HelpingAI/Dhanishtha-2.0-preview"
+        "description": "BilalRahib's TinyLLama NSFW Chatbot",
+        "hf_model_name": "bilalRahib/TinyLLama-NSFW-Chatbot"
     },
     "whisper-large-v3": {
         "task": "automatic-speech-recognition",
         "description": "openai whisper-large-v3",
         "hf_model_name": "openai/whisper-large-v3"
+    },
+    "Nusantara-4b-Indo-Chat": {
+        "task": "text-generation",
+        "description": "kalisai Nusantara-4b-Indo-Chat",
+        "hf_model_name": "kalisai/Nusantara-4b-Indo-Chat"
+    },
+   "cendol-llama2-7b-inst": {
+        "task": "text-generation",
+        "description": "indonlp cendol-llama2-7b-inst",
+        "hf_model_name": "indonlp/cendol-llama2-7b-inst"
+    },
+   "harry-potter-gpt2": {
+        "task": "text-generation",
+        "description": "akahana harry-potter-gpt2",
+        "hf_model_name": "akahana/harry-potter-gpt2"
+    },
+    "Mistral-7B-Sunda-v1.0": {
+        "task": "text-generation",
+        "description": "Nero10578 Mistral-7B-Sunda-v1.0",
+        "hf_model_name": "Nero10578/Mistral-7B-Sunda-v1.0"
+    },
+    "gpt2-indo-textgen": {
+        "task": "text-generation",
+        "description": "anugrahap gpt2-indo-textgen",
+        "hf_model_name": "anugrahap/gpt2-indo-textgen"
+    },
+    "cendol-mt5-small-inst": {
+        "task": "text-generation",
+        "description": "indonlp cendol-mt5-small-inst",
+        "hf_model_name": "indonlp/cendol-mt5-small-inst"
+    },
+    "madlad400-8b-lm": {
+        "task": "text-generation",
+        "description": "google madlad400-8b-lm",
+        "hf_model_name": "google/madlad400-8b-lm"
     }
 }
 
-# --- Penyimpanan Model Global (untuk Lazy Loading) ---
+# --- Lazy Loading ---
 models = {}
 
-# --- Fungsi Utility untuk Model Lazy Loading ---
+# --- Utility Lazy Loading ---
 def get_model_pipeline(model_name):
     """
     Memuat model hanya jika belum dimuat (lazy loading).
@@ -78,15 +126,14 @@ def get_model_pipeline(model_name):
 
         info = model_info[model_name]
         try:
-            # Gunakan 'hf_model_name' jika disediakan, jika tidak, gunakan model_name dengan prefix 'Lyon28/'
+
             hf_model_path = info.get("hf_model_name", f"Lyon28/{model_name}")
 
-            # Explicitly set device to "cpu" for CPU-only environments
             models[model_name] = pipeline(
                 info["task"],
                 model=hf_model_path,
-                device="cpu", # Penting: Pastikan ini "cpu" jika Anda tidak punya GPU
-                torch_dtype=torch.float32 # Tetap float32 untuk performa terbaik di CPU
+                device="cpu", 
+                torch_dtype=torch.float32
             )
             logger.info(f"✅ Model '{model_name}' (Path: {hf_model_path}) berhasil dimuat.")
         except Exception as e:
@@ -224,4 +271,4 @@ if __name__ == '__main__':
     # Untuk Hugging Face Spaces, port biasanya 7860
     # Menggunakan HOST dari environment variable jika tersedia, default ke 0.0.0.0
     # Debug=False untuk produksi
-    app.run(host=os.getenv('HOST', '0.0.0.0'), port=int(os.getenv('PORT', 7860)), debug=False)
+    app.run(host=os.getenv('HOST', '0.0.0.0'), port=int(os.getenv('PORT', 7860)), debug=False)