Update app.py

app.py

@@ -57,7 +57,8 @@ def generate_txt2img(current_model, prompt, is_negative=False, image_style="None
 
     API_TOKEN = os.environ.get("HF_READ_TOKEN")
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
-
+    if type(current_model) != type(""):
+        current_model = "stabilityai/stable-diffusion-2-1"
     api = InferenceApi(current_model)
     api.headers = headers
 
@@ -202,8 +203,8 @@ css = """
 }
 """
 
-#with gr.Blocks(css=css) as demo:
-with gr.Blocks() as demo:
+with gr.Blocks(css=css) as demo:
+    #with gr.Blocks() as demo:
     
     favicon = '<img src="" width="48px" style="display: inline">'
     gr.Markdown(