Update app.py

app.py

@@ -1,9 +1,8 @@
 import gradio as gr
 import numpy as np
 import random
-import spaces  # [uncomment to use ZeroGPU]
-from diffusers import DiffusionPipeline
 import torch
+from diffusers import DiffusionPipeline
 from tags import participant_tags, tribe_tags, skin_tone_tags, body_type_tags, tattoo_tags, piercing_tags, expression_tags, eye_tags, hair_style_tags, position_tags, fetish_tags, location_tags, camera_tags, atmosphere_tags
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -31,13 +30,12 @@ def update_model_version(version, state_model_repo_id):
     print(f"Model switched to {model_repo_id}")
     state_model_repo_id.set(model_repo_id)  # Update the state with the new model version
 
-@spaces.GPU  # [uncomment to use ZeroGPU]
+@gradio.GPU  # [uncomment to use ZeroGPU]
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
           selected_participant_tags, selected_tribe_tags, selected_skin_tone_tags, selected_body_type_tags,
           selected_tattoo_tags, selected_piercing_tags, selected_expression_tags, selected_eye_tags,
           selected_hair_style_tags, selected_position_tags, selected_fetish_tags, selected_location_tags,
           selected_camera_tags, selected_atmosphere_tags, active_tab, progress=gr.Progress(track_tqdm=True)):
-
     if active_tab == "Prompt Input":
         final_prompt = f'score_9, score_8_up, score_7_up, source_anime, {prompt}'
     else:
@@ -208,6 +206,8 @@ with gr.Blocks(css=css) as demo:
 
         with gr.Column(elem_id="right-column"):
             active_tab = gr.State("Prompt Input")
+            model_version_state = gr.State("v8")  # Default model version is v8
+
             with gr.Tabs() as tabs:
                 with gr.TabItem("Prompt Input") as prompt_tab:
                     prompt = gr.Textbox(
@@ -244,9 +244,9 @@ with gr.Blocks(css=css) as demo:
                 link_button_v11 = gr.Button("V11 Model", variant="primary")
 
             # Set the model version based on the button clicked
-            link_button_v7.click(update_model_version, inputs=["v7"], outputs=[])
-            link_button_v8.click(update_model_version, inputs=["v8"], outputs=[])
-            link_button_v11.click(update_model_version, inputs=["v11"], outputs=[])
+            link_button_v7.click(update_model_version, inputs=[gr.Text("v7")], outputs=[model_version_state])
+            link_button_v8.click(update_model_version, inputs=[gr.Text("v8")], outputs=[model_version_state])
+            link_button_v11.click(update_model_version, inputs=[gr.Text("v11")], outputs=[model_version_state])
 
         run_button.click(
             infer,