Update app.py

app.py

@@ -10,7 +10,7 @@ import replicate
 
 # Project by Nymbo
 
-# API_URL = "https://api-inference.huggingface.co/models/ovi054/rmx_flux"
+# API_URL = "https://api-inference.huggingface.co/models/codermert/mert2_flux"
 # API_TOKEN = os.getenv("HF_READ_TOKEN")
 # headers = {"Authorization": f"Bearer {API_TOKEN}"}
 # timeout = 100
@@ -22,7 +22,7 @@ def query(prompt, aspect_ratio="1:1", steps=28, cfg_scale=3.5, seed=-1, strength
     
     input = {
         "prompt": prompt,
-        "hf_lora": "ovi054/rmx_flux",
+        "hf_lora": "codermert/mert2_flux",
         "output_format": "jpg",
         "aspect_ratio": aspect_ratio,
         "num_inference_steps": steps,