Update app.py

app.py

@@ -5,15 +5,14 @@ import random
 from io import StringIO
 from datasets import load_dataset
 
-# Fetch dialogues from the provided URL
-response = requests.get("https://huggingface.co/datasets/Blessin/dialogues-one-liners/blob/main/dialogues_dataset.json")
-response.raise_for_status()  # Ensure the request was successful
-data = response.json()
-DIALOGUES = data["dialogues"]
+# Load the dataset from Hugging Face Spaces
+dataset = load_dataset("Blessin/dialogues-one-liners")
+DIALOGUES = dataset["train"]["dialogue"]
 
 def generate_statement():
-    # Fetch a random dialogue from the loaded dataset
-    return random.choice(DIALOGUES)
+    # Generate a random statement from the dataset
+    index = random.randint(0, len(DIALOGUES) - 1)
+    return DIALOGUES[index]
 
 def main():
     # Define the UI components using gr.Interface
@@ -29,4 +28,4 @@ def main():
     interface.launch(share=True)
 
 if __name__ == "__main__":
-    main()
+    main()