Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,20 +13,35 @@ pipe_image = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
|
|
13 |
|
14 |
def generate_headline(selected_language, text):
|
15 |
if selected_language == "Arabic":
|
16 |
-
translated_text =
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
image = pipe_image(english_headline).images[0]
|
21 |
return arabic_headline, image
|
22 |
-
|
23 |
elif selected_language == "English":
|
24 |
-
|
25 |
-
|
26 |
-
image = pipe_image(english_headline).images[0]
|
27 |
return english_headline, image
|
28 |
|
29 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
30 |
#First parameter is for the dropdown menu, and the second parameter is for the starter of the poem
|
31 |
["Arabic", "تعتبر انبعاثات الغازات الدفيئة، مثل ثاني أكسيد الكربون (CO2) والميثان (CH4)، من الأسباب الرئيسية لتغير المناخ العالمي. تؤدي الأنشطة البشرية، مثل حرق الوقود الأحفوري لإنتاج الطاقة وإزالة الغابات والعمليات الصناعية، إلى زيادة كبيرة في تركيز هذه الغازات في الغلاف الجوي. وفقًا للهيئة الحكومية الدولية المعنية بتغير المناخ (IPCC)، ارتفعت مستويات ثاني أكسيد الكربون بأكثر من 50٪ منذ عصر ما قبل الصناعة، مما ساهم في ارتفاع درجات الحرارة العالمية."]
|
32 |
]
|
@@ -42,7 +57,7 @@ interface = gr.Interface(
|
|
42 |
gr.Image(label="Generated Image")
|
43 |
],
|
44 |
|
45 |
-
examples=
|
46 |
)
|
47 |
interface.launch()
|
48 |
|
|
|
13 |
|
14 |
def generate_headline(selected_language, text):
|
15 |
if selected_language == "Arabic":
|
16 |
+
translated_text = translate_ar_to_en(text) # Translate Arabic to English
|
17 |
+
english_headline = generate_headline_english(translated_text) # Generate headline in English
|
18 |
+
arabic_headline = translate_en_to_ar(english_headline) # Translate headline back to Arabic
|
19 |
+
image = generate_image(english_headline) # Generate image based on English headline
|
|
|
20 |
return arabic_headline, image
|
21 |
+
|
22 |
elif selected_language == "English":
|
23 |
+
english_headline = generate_headline_english(text)
|
24 |
+
image = generate_image(english_headline)
|
|
|
25 |
return english_headline, image
|
26 |
|
27 |
+
def translate_ar_to_en(text):
|
28 |
+
var_ar_to_en = ar_to_en_translator(text)[0]['translation_text']
|
29 |
+
return var_ar_to_en
|
30 |
+
|
31 |
+
def translate_en_to_ar(text):
|
32 |
+
var_en_to_ar - en_to_ar_translator(text)[0]['translation_text']
|
33 |
+
return var_en_to_ar
|
34 |
+
|
35 |
+
def generate_headline_english(text):
|
36 |
+
result1 = headline_gen(text, max_length=100, truncation=True)
|
37 |
+
result2 = result1[0]['generated_text']
|
38 |
+
return result2
|
39 |
+
|
40 |
+
def generate_image(text):
|
41 |
+
my_image = pipe_image(text).images[0]
|
42 |
+
return my_image
|
43 |
+
|
44 |
+
examples = [
|
45 |
#First parameter is for the dropdown menu, and the second parameter is for the starter of the poem
|
46 |
["Arabic", "تعتبر انبعاثات الغازات الدفيئة، مثل ثاني أكسيد الكربون (CO2) والميثان (CH4)، من الأسباب الرئيسية لتغير المناخ العالمي. تؤدي الأنشطة البشرية، مثل حرق الوقود الأحفوري لإنتاج الطاقة وإزالة الغابات والعمليات الصناعية، إلى زيادة كبيرة في تركيز هذه الغازات في الغلاف الجوي. وفقًا للهيئة الحكومية الدولية المعنية بتغير المناخ (IPCC)، ارتفعت مستويات ثاني أكسيد الكربون بأكثر من 50٪ منذ عصر ما قبل الصناعة، مما ساهم في ارتفاع درجات الحرارة العالمية."]
|
47 |
]
|
|
|
57 |
gr.Image(label="Generated Image")
|
58 |
],
|
59 |
|
60 |
+
examples=examples
|
61 |
)
|
62 |
interface.launch()
|
63 |
|