Update app.py

app.py

@@ -2,21 +2,9 @@
 import os
 import gradio as gr
 from transformers import pipeline
-
-# Load the API key from environment variables
-api_key = os.getenv("HF_API_KEY")
-
-if api_key is None:
-    print("HF_API_KEY not found. Please set it in the Space's secrets.")
-else:
-    # Set the necessary environment variables for Hugging Face Transformers
-    os.environ["HF_HOME"] = "/workspace"  # Use the /workspace directory in Hugging Face Spaces
-    os.environ["TRANSFORMERS_CACHE"] = "/workspace/cache"  # Cache directory in Space
-    os.environ["HF_API_KEY"] = api_key  # Set the API key for Transformers library
-
     # Initialize the model using the API key and Transformers pipeline
     try:
-        model_name = "tiiuae/falcon-180B"  # Ensure this is the correct model name
+        model_name = "tiiuae/falcon-40b-instruct"  # Ensure this is the correct model name
         nlp_model = pipeline('text-generation', model=model_name)
         print("Model loaded successfully.")
     except Exception as e: