change space for hw5

app.py

@@ -10,8 +10,13 @@ from diffusers import DiffusionPipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
 model_repo_id = "CompVis/stable-diffusion-v1-4"
-model_dropdown = ['stabilityai/sdxl-turbo', 'CompVis/stable-diffusion-v1-4' ]
+model_dropdown = ["stabilityai/sdxl-turbo", "CompVis/stable-diffusion-v1-4"]
 
+models = [
+    "gstranger/kawaiicat-lora-1.4",
+    "CompVis/stable-diffusion-v1-4",
+    "stabilityai/sdxl-turbo",
+]
 
 
 if torch.cuda.is_available():
@@ -28,6 +33,7 @@ MAX_IMAGE_SIZE = 1024
 
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
+    model_id,
     prompt,
     negative_prompt,
     randomize_seed,
@@ -44,7 +50,12 @@ def infer(
 
     generator = torch.Generator().manual_seed(seed)
 
-    pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+    pipe = DiffusionPipeline.from_pretrained(
+        model_id,
+        torch_dtype=torch_dtype,
+        requires_safety_checker=False,
+        safety_checker=None,
+    )
     pipe = pipe.to(device)
 
     image = pipe(
@@ -60,11 +71,9 @@ def infer(
     return image, seed
 
 
-
 examples = [
-    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
-    "An astronaut riding a green horse",
-    "A delicious ceviche cheesecake slice",
+    "kawaiicat. The cat is sitting. The cat's tail is curled up at the end. The cat is pleased and is enjoying its time.",
+    "kawaiicat. The cat is sitting upright. The cat is eating some noodles with the chopsticks from a green bowl, which it's holding in his hands.",
 ]
 
 css = """
@@ -76,7 +85,22 @@ css = """
 
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(" # Text-to-Image SemaSci Template")
+        gr.Markdown(" # Text-to-Image kawaiicat Stickers")
+        with gr.Row():
+            # Dropdown to select the model from Hugging Face
+            model_id = gr.Dropdown(
+                label="Model",
+                choices=models,
+                value=models[0],  # Default model
+            )
+
+        lora_scale = gr.Slider(
+            label="LORA Scale",
+            minimum=0,
+            maximum=1,
+            step=0.01,
+            value=1,
+        )
 
         with gr.Row():
             prompt = gr.Text(
@@ -92,22 +116,22 @@ with gr.Blocks(css=css) as demo:
         result = gr.Image(label="Result", show_label=False)
 
         with gr.Accordion("Advanced Settings", open=False):
-#            model_repo_id = gr.Text(
-#                label="Model Id",
-#                max_lines=1,
-#                placeholder="Choose model",
-#                visible=True,
-#                value=model_repo_id,
-#            )
-            model_repo_id = gr.Dropdown(
-                label="Model Id",
-                choices=model_dropdown,
-                info="Choose model",
-                visible=True,
-                allow_custom_value=True,
-                value=model_repo_id,
-            )            
-            
+            #            model_repo_id = gr.Text(
+            #                label="Model Id",
+            #                max_lines=1,
+            #                placeholder="Choose model",
+            #                visible=True,
+            #                value=model_repo_id,
+            #            )
+            # model_id = gr.Dropdown(
+            #     label="Model Id",
+            #     choices=models,
+            #     info="Choose model",
+            #     visible=True,
+            #     allow_custom_value=True,
+            #     value=models,
+            # )
+
             negative_prompt = gr.Text(
                 label="Negative prompt",
                 max_lines=1,
@@ -164,6 +188,7 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit],
         fn=infer,
         inputs=[
+            model_id,
             prompt,
             negative_prompt,
             randomize_seed,
@@ -179,4 +204,3 @@ with gr.Blocks(css=css) as demo:
 
 if __name__ == "__main__":
     demo.launch()
-