Update app.py

app.py

@@ -1,22 +1,22 @@
-import gradio as gr
+import requests
 
-# Identifier of the Hugging Face Space
-space_url = "Canstralian/WhiteRabbitNeo"
+# Replace with the correct Space API URL
+api_url = "https://hf.space/embed/Canstralian/WhiteRabbitNeo/api/predict/"
 
-# Connect to the Space
-interface = gr.Interface.load(space_url)
+# The input payload to send to the model
+payload = {"data": ["A new bug bounty to solve"]}
 
-# Prompt to generate code and sentences
-prompt = "A new bug bounty to solve"
-
-# Get the generated result
 try:
-    result = interface(prompt)  # Ensure the interface supports string input and output
-    print("Generated Result:", result)
-
+    # Make a POST request to the API
+    response = requests.post(api_url, json=payload)
+    response.raise_for_status()  # Raise an exception for HTTP errors
+    
+    # Parse the result
+    result = response.json()  # Parse the JSON response from the API
+    print("Generated Result:", result["data"][0])  # Access the generated result
+    
     # Optionally, save the result to a text file
     with open("generated_output.txt", "w") as f:
-        f.write(result)
-
-except Exception as e:
+        f.write(result["data"][0])
+except requests.exceptions.RequestException as e:
     print("An error occurred:", e)