Update app.py

app.py

@@ -1,11 +1,3 @@
-
-
-# # initialize the client
-# client = OpenAI(
-#   base_url="https://api-inference.huggingface.co/v1",
-#   api_key=os.environ.get('')#"hf_xxx" # Replace with your token
-# ) 
-
 import numpy as np
 import streamlit as st
 from openai import OpenAI
@@ -31,8 +23,8 @@ model_links = {
     "Meta-Llama-3.1-405B-Instruct-FP8": "meta-llama/Meta-Llama-3.1-405B-Instruct-FP8",
     "Meta-Llama-3.1-405B-Instruct": "meta-llama/Meta-Llama-3.1-405B-Instruct",
     "Mistral-Nemo-Instruct-2407": "mistralai/Mistral-Nemo-Instruct-2407",
-    "Meta-Llama-3-70B-Instruct": "meta-llama/Meta-Llama-3-70B-Instruct",
-    "Meta-Llama-3-8B-Instruct": "meta-llama/Meta-Llama-3-8B-Instruct",
+    "Text-to-IMG-FLUX.1-dev": "black-forest-labs/FLUX.1-dev",
+    "Text-to-IMG-NSFW-gen-v2": "UnfilteredAI/NSFW-gen-v2",
     "C4ai-command-r-plus": "CohereForAI/c4ai-command-r-plus",
     "Aya-23-35B": "CohereForAI/aya-23-35B",
     "Zephyr-orpo-141b-A35b-v0.1": "HuggingFaceH4/zephyr-orpo-141b-A35b-v0.1",
@@ -42,7 +34,7 @@ model_links = {
     "Yi-1.5-34B-Chat": "01-ai/Yi-1.5-34B-Chat",
     "Gemma-2-27b-it": "google/gemma-2-27b-it",
     "Meta-Llama-2-70B-Chat-HF": "meta-llama/Llama-2-70b-chat-hf",
-    "Meta-Llama-2-7B-Chat-HF": "meta-llama/Llama-2-7b-chat-hf",
+    "Text-to-IMG-ByteDance/SDXL-Lightning": "ByteDance/SDXL-Lightning",
     "Meta-Llama-2-13B-Chat-HF": "meta-llama/Llama-2-13b-chat-hf",
     "Mistral-7B-Instruct-v0.1": "mistralai/Mistral-7B-Instruct-v0.1",
     "Mistral-7B-Instruct-v0.2": "mistralai/Mistral-7B-Instruct-v0.2",