Update app.py

app.py

@@ -25,7 +25,7 @@ def load_stories():
 # Load system prompts from Google Sheets
 def load_prompts():
     prompts_data = prompts_sheet.get_all_values()
-    prompts = [prompt[0] for prompt in prompts_data if prompt[0] != "System Prompt"]  # Skip header row
+    prompts = [prompt[0] for prompt in prompts_data[1:] if prompt[0] != "System Prompt"]  # Skip header row
     return prompts
 
 # Load available stories and prompts
@@ -73,7 +73,7 @@ chat_history = []
 def interact(user_input, history):
     global tokenizer, model
     try:
-        if tokenizer is None or model is None:
+        if tokenizer is None or model is not initialized:
             raise ValueError("Tokenizer or model is not initialized.")
         
         messages = history + [{"role": "user", "content": user_input}]
@@ -97,7 +97,7 @@ def interact(user_input, history):
         formatted_history = [(entry["content"], None) if entry["role"] == "user" else (None, entry["content"]) for entry in history if entry["role"] in ["user", "assistant"]]
         return "", formatted_history, history
     except Exception as e:
-        if torch.cuda.is_available():
+        if torch.cuda.is available():
             torch.cuda.empty_cache()
         print(f"Error during interaction: {e}")
         raise gr.Error(f"An error occurred during interaction: {str(e)}")