Update app.py

app.py

@@ -9,7 +9,7 @@ headline_gen = pipeline("text2text-generation", model="Michau/t5-base-en-generat
 ar_to_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ar-en")
 en_to_ar_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ar")
 
-image_gen = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1")
+pipe_image = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
 
 def generate_headline(selected_language, text):
     if selected_language == "Arabic":
@@ -17,13 +17,13 @@ def generate_headline(selected_language, text):
         result1 = headline_gen(translated_text, max_length=100, truncation=True) #Translated text will go to the model, and return the proper heading in english
         english_headline = result1[0]['generated_text'] #The heading in english
         arabic_headline = en_to_ar_translator(english_headline)[0]['translation_text'] #Translate the heading to Arabic
-        image = image_gen(english_headline).images[0]
+        image = pipe_image(english_headline).images[0]
         return arabic_headline, image
 
     elif selected_language == "English":
         result1 = headline_gen(text, max_length=100, truncation=True)
         english_headline = result1[0]['generated_text']
-        image = image_gen(english_headline).images[0]
+        image = pipe_image(english_headline).images[0]
         return english_headline, image
 
 ex = [