Update app.py

app.py

@@ -1,9 +1,26 @@
 import gradio as gr
 import random
+import requests
 
 def generate_response(input, history):
-    #Hit the API and get results
-    bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
+    #bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
+
+    #Hit the Sagemaker API and get results
+    
+    url = "https://runtime.sagemaker.eu-west-1.amazonaws.com/endpoints/jumpstart-dft-meta-textgeneration-llama-2-7b-f/invocations"
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": "Bearer <TOKEN>",  # Replace with your actual token
+    }
+    payload = {"key": "value"}
+    
+    response = requests.post(url, json=payload, headers=headers)
+    
+    if response.status_code == 200:
+        print("Request was successful:", response.json())
+    else:
+        print("Request failed:", response.status_code, response.text)
+
 
     return bot_message