Update app.py

app.py

@@ -1,16 +1,19 @@
+import os
 import streamlit as st
 from clarifai.client.model import Model
 
+# Read API key from environment variable
+api_key = os.getenv("CLARIFAI_PAT")
+
 # Function to get prediction from the model
 def get_model_prediction(prompt):
     model_url = "https://clarifai.com/mistralai/completion/models/codestral-22b-instruct"
-    pat = "CodestralPat"
-    model = Model(url=model_url, pat=pat)
+    model = Model(url=model_url, pat=api_key)
     model_prediction = model.predict_by_bytes(prompt.encode(), input_type="text")
     return model_prediction.outputs[0].data.text.raw
 
 # Streamlit interface
-st.title("Codestral on Clarifai")
+st.title("AI Model Prediction with Clarifai")
 st.write("Enter a prompt and get a prediction from the Clarifai model.")
 
 prompt = st.text_input("Enter your prompt:", "What's the future of AI?")