add model for indonesian stories

app.py

@@ -21,6 +21,9 @@ MODELS = {
     "GPT-2 Small finetuned on Indonesian academic journals": {
         "url": "https://api-inference.huggingface.co/models/Galuh/id-journal-gpt2"
     },
+    "GPT-2 Small finetuned on Indonesian stories": {
+        "url": "https://api-inference.huggingface.co/models/cahya/gpt2-small-indonesian-story"
+    },
 }
 
 def get_image(text: str):
@@ -128,6 +131,8 @@ if model_name in ["GPT-2 Small", "GPT-2 Medium"]:
     prompt_group_name = "GPT-2"
 elif model_name in ["GPT-2 Small finetuned on Indonesian academic journals"]:
     prompt_group_name = "Indonesian Journals"
+elif model_name in ["GPT-2 Small finetuned on Indonesian stories"]:
+    prompt_group_name = "Indonesian Stories"
 
 session_state = SessionState.get(prompt=None, prompt_box=None, text=None)
 

prompts.py

@@ -32,5 +32,9 @@ PROMPT_LIST = {
         "Psikologi (psychology)": ["Penelitian ini bertujuan untuk mengetahui perilaku wirausaha remaja yang diprediksi dari motivasi intrinsik", "Tujuan dari penelitian ini adalah untuk mendapatkan data empiris mengenai gambaran peta bakat mahasiswa Fakultas Psikologi Unjani"],
         "Ekonomi (economics)": ["Faktor kepuasan dan kepercayaan konsumen merupakan dua faktor kunci dalam meningkatkan penetrasi e-commerce. Penelitian yang dilakukan", "Penelitian ini bertujuan untuk menganalisis pola konsumsi pangan di Indonesia", "Model GTAP diimplementasikan untuk melihat dampak yang ditimbulkan pada PDB"],
         "Teknologi Informasi (IT)": ["pembuatan aplikasi ini menggunakan pengembangan metode Waterfall dan dirancang mengguynakan Unified Modeling Language (UML) dengan bahasa pemrograman", "Berdasarkan masalah tersebut, maka penulis termotivasi untuk membangun Pengembangan Sistem Informasi Manajemen"]
+    },
+    "Indonesian Stories": {
+        "Cerita Cinta (Love Story)": ["Pertama aku ingin kamu tahu, tidak ada satu pun dari masa lalu yang kita lewati bersama pernah aku sesali.", "Bersamamu aku pernah merasakan mencintai dan dicintai. Kamu pernah menjadi tempat aku bersandar.", "Aku tenggelam dalam lautan rumput liar yang lebih tinggi dari batas lutut orang dewasa sambil memandangi langit sore yang kelabu"],
+        "Cerita Horor (Horor)": ["Seketika jantungku berhenti berdetak mendengar desahan suara", "Rumah itu sudah lama kosong, ditinggalkan penghuninya."],
     }
 }