Update app.py

app.py

@@ -20,6 +20,8 @@ os.environ["TRAINER_TELEMETRY"]= "0"
 # Constants
 
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
+BASETEN_API = os.environ.get("BASETEN_API", None)
+BASETEN_KEY = os.environ.get("BASETEN_KEY", None)
 
 # Set the device to GPU or CPU
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -194,14 +196,14 @@ with rl:
             st.error("Please record audio or upload a file to transcribe.")
 
     # End of Box 3 and third Carousel Item
-    st.markdown('''<h3><i class="fa fa-pencil"></i> Form 3</h3>''', unsafe_allow_html=True)
+    st.markdown('''<h3><i class="fa fa-pencil"></i> Chat with Mistral</h3>''', unsafe_allow_html=True)
 with right:
     # Box 4: Form 3
     prompt3 = st.text_input("Enter Prompt", key="prompt3")
-    image_url3 = st.text_input("Enter Image URL", key="image_url3")
+    #image_url3 = st.text_input("Enter Image URL", key="image_url3")
     if st.button("Submit Form 3", key="submit3"):
-        payload = {"prompt": prompt3, "image_url": image_url3}
-        response = requests.post("https://d3ndnam3-hf.space/api", json=payload)
+        payload = {"prompt": prompt3}
+        response = requests.post("{BASETEN_API}", headers={"Authorization": "Api-Key {BASETEN_KEY}"}, json=payload)
         if response.status_code == 200:
             st.write(f"**Response:** {response.json().get('response', 'No response')}")
         else: