Update model_api.py

model_api.py

@@ -5,7 +5,9 @@ import re
 
 def query_model(prompt):
     try:
-        HF_TOKEN = os.getenv("HF_TOKEN")
+        import streamlit as st
+
+        HF_TOKEN = st.secrets["HF_TOKEN"]
 
         client = InferenceClient(
             provider="auto",