Update app.py

app.py

@@ -11,6 +11,7 @@ from peft import PeftModel, LoraConfig, get_peft_model, prepare_model_for_int8_t
 model_dict = {
     "Stable Diffusion": {
         "Models": [
+            "stabilityai/stable-diffusion-3-medium"
             "CompVis/stable-diffusion-v1-4",
             "stabilityai/stable-diffusion-2-1",
             "runwayml/stable-diffusion-v1-5",
@@ -140,7 +141,7 @@ if model_type == "Stable Diffusion":
         for i, image in enumerate(images):
             cols[i].image(image, caption=f"Image {i+1}", use_column_width=True)
 
-elif model_type == "Llama":
+if model_type == "Llama":
     model = st.selectbox("Select a Llama model:", model_dict[model_type]["Models"])
     tokenizer = st.selectbox("Select a tokenizer:", model_dict[model_type]["Tokenizers"])
     feature = st.selectbox("Select a feature:", model_dict[model_type]["Features"])