Update app.py

app.py

@@ -1,16 +1,10 @@
-import os
-from huggingface_hub import login
 import torch
+from huggingface_hub import login
 from transformers import pipeline
 import gradio as gr
 
-# Set the Hugging Face token from the environment variable
-hf_token = os.getenv("HF_TOKEN")
-if hf_token is None:
-    raise ValueError("Hugging Face token is not set in the environment variable.")
-
-# Log in to Hugging Face with the token
-login(token=hf_token)
+# Log in to Hugging Face (this will use the token stored from huggingface-cli login)
+login()
 
 # Define the model ID
 model_id = "meta-llama/Llama-3.2-1B-Instruct"
@@ -41,4 +35,4 @@ iface = gr.Interface(
 )
 
 # Launch the Gradio app
-iface.launch()
+iface.launch()