prithivMLmods commited on
Commit
952d9f0
·
verified ·
1 Parent(s): 3e5ce53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -65,7 +65,7 @@ def progress_bar_html(label: str) -> str:
65
  # ---------------------------
66
  # Uses a text-only model: FastThink-0.5B-Tiny
67
 
68
- model_id_text = "prithivMLmods/FastThink-0.5B-Tiny"
69
  tokenizer = AutoTokenizer.from_pretrained(model_id_text)
70
  model = AutoModelForCausalLM.from_pretrained(
71
  model_id_text,
@@ -331,9 +331,6 @@ with gr.Blocks(css=".gradio-container {max-width: 900px; margin: auto;}") as dem
331
  run_img_btn.click(generate_image_lora,
332
  inputs=[prompt_img, negative_prompt_img, use_neg_checkbox, seed_slider, width_slider, height_slider, guidance_slider, randomize_seed_checkbox, style_radio, lora_dropdown],
333
  outputs=[img_output, seed_output])
334
-
335
- gr.Markdown("### Adjustments")
336
- gr.Markdown("Each tab has been implemented separately. Feel free to adjust parameters and layout as needed in each tab.")
337
-
338
  if __name__ == "__main__":
339
  demo.queue(max_size=20).launch(share=True)
 
65
  # ---------------------------
66
  # Uses a text-only model: FastThink-0.5B-Tiny
67
 
68
+ model_id_text = "prithivMLmods/DeepHermes-3-Llama-3-3B-Preview-abliterated"
69
  tokenizer = AutoTokenizer.from_pretrained(model_id_text)
70
  model = AutoModelForCausalLM.from_pretrained(
71
  model_id_text,
 
331
  run_img_btn.click(generate_image_lora,
332
  inputs=[prompt_img, negative_prompt_img, use_neg_checkbox, seed_slider, width_slider, height_slider, guidance_slider, randomize_seed_checkbox, style_radio, lora_dropdown],
333
  outputs=[img_output, seed_output])
334
+
 
 
 
335
  if __name__ == "__main__":
336
  demo.queue(max_size=20).launch(share=True)