salomonsky commited on
Commit
520d65e
verified
1 Parent(s): 986e29b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,7 +46,7 @@ def enhance_prompt(text, client=client):
46
  try:
47
  enhanced = client.text_generation(
48
  "Generate a photorealistic, detailed txt2img prompt: " + text,
49
- model="deepseek-ai/Janus-Pro-7B",)
50
  return enhanced[:200]
51
 
52
  except Exception as e:
@@ -179,7 +179,7 @@ def main():
179
 
180
  prompt = st.sidebar.text_area("Descripci贸n de la imagen", height=150, max_chars=500)
181
  format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9", "1:1"])
182
- model_option = st.sidebar.selectbox("Modelo", ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-dev"])
183
  prompt_enhance = st.sidebar.checkbox("Mejorar Prompt", True)
184
  num_variants = st.sidebar.slider("N煤mero de im谩genes", 1, 8, 8)
185
  width, height = (720, 1280) if format_option == "9:16" else (1280, 720) if format_option == "16:9" else (1280, 1280)
 
46
  try:
47
  enhanced = client.text_generation(
48
  "Generate a photorealistic, detailed txt2img prompt: " + text,
49
+ model="mistralai/Mixtral-8x7B-Instruct-v0.1",)
50
  return enhanced[:200]
51
 
52
  except Exception as e:
 
179
 
180
  prompt = st.sidebar.text_area("Descripci贸n de la imagen", height=150, max_chars=500)
181
  format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9", "1:1"])
182
+ model_option = st.sidebar.selectbox("Modelo", ["black-forest-labs/FLUX.1-schnell"])
183
  prompt_enhance = st.sidebar.checkbox("Mejorar Prompt", True)
184
  num_variants = st.sidebar.slider("N煤mero de im谩genes", 1, 8, 8)
185
  width, height = (720, 1280) if format_option == "9:16" else (1280, 720) if format_option == "16:9" else (1280, 1280)