Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,35 +9,34 @@ import torch
|
|
9 |
accelerator = Accelerator()
|
10 |
|
11 |
# Inisialisasi model terjemahan
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
model = MarianMTModel.from_pretrained(model_name)
|
16 |
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
st.session_state.model_initialized = True
|
31 |
-
st.markdown("<span style='color: green'>Model dijalankan di CPU dengan Accelerate.</span>", unsafe_allow_html=True)
|
32 |
-
else:
|
33 |
-
# Jika tidak menggunakan accelerate, tetap di CPU
|
34 |
-
pipe = pipe.to("cpu")
|
35 |
st.session_state.model_initialized = True
|
36 |
-
st.markdown("<span style='color:
|
|
|
|
|
|
|
|
|
|
|
37 |
|
38 |
|
39 |
st.title("AI Diffusion Bahasa Indonesia")
|
40 |
-
st.write("
|
41 |
text = st.text_area('Masukkan teks dalam bahasa Indonesia')
|
42 |
|
43 |
if st.button("Proses"):
|
@@ -54,4 +53,4 @@ if st.button("Proses"):
|
|
54 |
# Tampilkan hasil
|
55 |
st.image(image, caption='Hasil Diffusion User', use_column_width=True)
|
56 |
else:
|
57 |
-
st.write("Silakan masukkan teks terlebih dahulu.")
|
|
|
9 |
accelerator = Accelerator()
|
10 |
|
11 |
# Inisialisasi model terjemahan
|
12 |
+
model_name = "Helsinki-NLP/opus-mt-id-en"
|
13 |
+
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
14 |
+
model = MarianMTModel.from_pretrained(model_name)
|
|
|
15 |
|
16 |
+
# Instalisasi Diffusion User
|
17 |
+
model_id = "runwayml/stable-diffusion-v1-5"
|
18 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
|
19 |
|
20 |
+
if torch.cuda.is_available():
|
21 |
+
# Pipe akan dipindahkan ke GPU jika tersedia
|
22 |
+
pipe = pipe.to(accelerator.device)
|
23 |
+
st.session_state.model_initialized = True
|
24 |
+
st.markdown("<span style='color: green'>Model dipindahkan ke GPU.</span>", unsafe_allow_html=True)
|
25 |
+
else:
|
26 |
+
# Memeriksa apakah menggunakan accelerate di CPU
|
27 |
+
if accelerator.device.type == 'cpu':
|
28 |
+
with accelerator.device:
|
|
|
|
|
|
|
|
|
|
|
29 |
st.session_state.model_initialized = True
|
30 |
+
st.markdown("<span style='color: green'>Model dijalankan di CPU dengan Accelerate.</span>", unsafe_allow_html=True)
|
31 |
+
else:
|
32 |
+
# Jika tidak menggunakan accelerate, tetap di CPU
|
33 |
+
pipe = pipe.to("cpu")
|
34 |
+
st.session_state.model_initialized = True
|
35 |
+
st.markdown("<span style='color: red'>GPU tidak ditemukan. Model tetap di CPU.</span>", unsafe_allow_html=True)
|
36 |
|
37 |
|
38 |
st.title("AI Diffusion Bahasa Indonesia")
|
39 |
+
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")
|
40 |
text = st.text_area('Masukkan teks dalam bahasa Indonesia')
|
41 |
|
42 |
if st.button("Proses"):
|
|
|
53 |
# Tampilkan hasil
|
54 |
st.image(image, caption='Hasil Diffusion User', use_column_width=True)
|
55 |
else:
|
56 |
+
st.write("Silakan masukkan teks terlebih dahulu.")
|