Update app.py

app.py

@@ -9,35 +9,34 @@ import torch
 accelerator = Accelerator()
 
 # Inisialisasi model terjemahan
-if "model_initialized" not in st.session_state:
-    model_name = "Helsinki-NLP/opus-mt-id-en"
-    tokenizer = MarianTokenizer.from_pretrained(model_name)
-    model = MarianMTModel.from_pretrained(model_name)
+model_name = "Helsinki-NLP/opus-mt-id-en"
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
 
-    # Instalisasi Diffusion User
-    model_id = "runwayml/stable-diffusion-v1-5"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+# Instalisasi Diffusion User
+model_id = "runwayml/stable-diffusion-v1-5"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
 
-    if torch.cuda.is_available():
-        # Pipe akan dipindahkan ke GPU jika tersedia
-        pipe = pipe.to(accelerator.device)
-        st.session_state.model_initialized = True
-        st.markdown("<span style='color: green'>Model dipindahkan ke GPU.</span>", unsafe_allow_html=True)
-    else:
-        # Memeriksa apakah menggunakan accelerate di CPU
-        if accelerator.device.type == 'cpu':
-            with accelerator.device:
-                st.session_state.model_initialized = True
-                st.markdown("<span style='color: green'>Model dijalankan di CPU dengan Accelerate.</span>", unsafe_allow_html=True)
-        else:
-            # Jika tidak menggunakan accelerate, tetap di CPU
-            pipe = pipe.to("cpu")
+if torch.cuda.is_available():
+    # Pipe akan dipindahkan ke GPU jika tersedia
+    pipe = pipe.to(accelerator.device)
+    st.session_state.model_initialized = True
+    st.markdown("<span style='color: green'>Model dipindahkan ke GPU.</span>", unsafe_allow_html=True)
+else:
+    # Memeriksa apakah menggunakan accelerate di CPU
+    if accelerator.device.type == 'cpu':
+        with accelerator.device:
             st.session_state.model_initialized = True
-            st.markdown("<span style='color: red'>GPU tidak ditemukan. Model tetap di CPU.</span>", unsafe_allow_html=True)
+            st.markdown("<span style='color: green'>Model dijalankan di CPU dengan Accelerate.</span>", unsafe_allow_html=True)
+    else:
+        # Jika tidak menggunakan accelerate, tetap di CPU
+        pipe = pipe.to("cpu")
+        st.session_state.model_initialized = True
+        st.markdown("<span style='color: red'>GPU tidak ditemukan. Model tetap di CPU.</span>", unsafe_allow_html=True)
 
 
 st.title("AI Diffusion Bahasa Indonesia")
-st.write("ini adalah pembuat gambar menggunakan AI, saya menggunakan model runwayml/stable-diffusion-v1.5 dan Helsinki-NLP/opus-mt-id-e. berkerja dengan baik, tetapi karna saya menggunakan kapasitas geratisan jadi prosesnya akan sangat lama")
+st.write("Ini adalah pembuat gambar menggunakan AI, saya menggunakan model runwayml/stable-diffusion-v1.5 dan Helsinki-NLP/opus-mt-id-e. Berkerja dengan baik, tetapi karena saya menggunakan kapasitas geratisan jadi prosesnya akan sangat lama")
 text = st.text_area('Masukkan teks dalam bahasa Indonesia')
 
 if st.button("Proses"):
@@ -54,4 +53,4 @@ if st.button("Proses"):
             # Tampilkan hasil
             st.image(image, caption='Hasil Diffusion User', use_column_width=True)
     else:
-        st.write("Silakan masukkan teks terlebih dahulu.")
+        st.write("Silakan masukkan teks terlebih dahulu.")