Update app.py

app.py

@@ -12,10 +12,10 @@ translator = pipeline("translation", model="Helsinki-NLP/opus-mt-id-en")
 terjemahkan = pipeline("translation", model="Helsinki-NLP/opus-mt-en-id")
 
 # Menggunakan pipa untuk klasifikasi diagnosis penyakit
-disease_classifier = pipeline("text-classification", model="BenK10/disease-diagnosis")
+# disease_classifier = pipeline("text-classification", model="BenK10/disease-diagnosis")
 
 # Pipeline generasi teks dengan format pesan
-text_generation = pipeline("text-generation", model="SumayyaAli/tiny-llama-1.1b-chat-medical")
+text_generation = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0")
 
 def preprocess_text(teks):
     # Pra-pemrosesan teks menggunakan NLTK
@@ -36,21 +36,25 @@ def predict_disease(gejala):
         terjemahan_inggris = terjemahan[0]["translation_text"]
 
         # Klasifikasi diagnosis penyakit dengan fitur tambahan
-        klasifikasi = disease_classifier(terjemahan_inggris)
+        # klasifikasi = disease_classifier(terjemahan_inggris)
 
         # Mengembalikan diagnosis dan skor kepercayaan
-        diagnosis = klasifikasi[0]["label"]
-        kepercayaan = klasifikasi[0]["score"]
+        # diagnosis = klasifikasi[0]["label"]
+        # kepercayaan = klasifikasi[0]["score"]
         
         # Menggunakan format pesan untuk definisi penyakit
-        definisi = text_generation([{"role": "user", "content": f"What is {diagnosis}?"}])
+        definisi = text_generation([{
+            "role": "user", 
+            "content": f"You are a doctor who needs to diagnose a patient's illness. In your opinion, if I a have a {terjemahan_inggris}, what illness could it be?"
+        }])
         definisi_teks = definisi[0]['generated_text']
         
         # Melakukan terjemahan definisi kembali ke bahasa Indonesia
         terjemahan_definisi = terjemahkan(definisi_teks, max_length=100)
         definisi = terjemahan_definisi[0]["translation_text"]
         
-        return diagnosis, kepercayaan, definisi
+        # return diagnosis, kepercayaan, definisi
+        return definisi
     
     except Exception as e:
         return "Tidak dapat mendiagnosis", 0.0, "Terjadi kesalahan saat mendiagnosis penyakit."
@@ -77,8 +81,8 @@ with st.form("diagnosis_form"):
         if gejala and usia and jenis_kelamin:
             diagnosis, kepercayaan, definisi = predict_disease(gejala)
             if diagnosis != "Tidak dapat mendiagnosis":
-                st.success(f"Diagnosis: {diagnosis}")
-                st.write(f"Dengan tingkat akurasi: {kepercayaan:.2f}")
+                # st.success(f"Diagnosis: {diagnosis}")
+                # st.write(f"Dengan tingkat akurasi: {kepercayaan:.2f}")
                 st.write(f"{definisi}")
             else:
                 st.error(definisi)