SarahMarzouq commited on
Commit
d65550f
·
verified ·
1 Parent(s): 4951384

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -11
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 = ar_to_en_translator(text)[0]['translation_text'] #Translate Arabic text to english, to use the model
17
- result1 = headline_gen(translated_text, max_length=100, truncation=True) #Translated text will go to the model, and return the proper heading in english
18
- english_headline = result1[0]['generated_text'] #The heading in english
19
- arabic_headline = en_to_ar_translator(english_headline)[0]['translation_text'] #Translate the heading to Arabic
20
- image = pipe_image(english_headline).images[0]
21
  return arabic_headline, image
22
-
23
  elif selected_language == "English":
24
- result1 = headline_gen(text, max_length=100, truncation=True)
25
- english_headline = result1[0]['generated_text']
26
- image = pipe_image(english_headline).images[0]
27
  return english_headline, image
28
 
29
- ex = [
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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=ex
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