Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import os
 import torch
 from transformers import pipeline
 from diffusers import StableDiffusionPipeline
@@ -25,6 +26,7 @@ else:
     title = "Txt to Image by CPU"
     pipeline = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=token)
 
+title = os.environ['model_fetch']
 
 def predict(prompt):
   predictions = pipeline(prompt, guidance_scale=7.5).images[0]