Update app.py

app.py

@@ -1,21 +1,21 @@
-import torch
 import spaces
 import os
 import uuid
 import time
 import json
+import torch
 from datetime import datetime, timedelta
 from threading import Thread
 
 # Gradio and HuggingFace imports
 import gradio as gr
-from gradio.themes import Base
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 from datasets import Dataset
 from huggingface_hub import HfApi, login
 
 # Model configuration
 checkpoint = "WillHeld/soft-raccoon"
+
 # Set device based on availability
 if torch.cuda.is_available():
     device = "cuda"
@@ -24,7 +24,7 @@ else:
     print("CUDA not available, using CPU instead. This will be much slower.")
 
 # Dataset configuration
-DATASET_NAME = "your-username/soft-raccoon-conversations"  # Change to your username
+DATASET_NAME = "WillHeld/soft-raccoon-conversations"  # Change to your username
 SAVE_INTERVAL_MINUTES = 5  # Save data every 5 minutes
 last_save_time = datetime.now()
 
@@ -38,7 +38,7 @@ conversations = []
 
 # Hugging Face authentication
 # Uncomment this line to login with your token
-login(token=os.environ.get("HF_TOKEN"))
+# login(token=os.environ.get("HF_TOKEN"))
 
 
 def save_to_dataset():