Update app.py

app.py

@@ -2,6 +2,10 @@ import gradio as gr
 import transformers
 import torch
 
+import os
+
+hf_key = os.getenv("HF_TOKEN")
+
 # Initialize the model
 model_id = "bmi-labmedinfo/Igea-350M-v0.0.1"
 pipeline = transformers.pipeline(
@@ -9,6 +13,7 @@ pipeline = transformers.pipeline(
     model=model_id,
     model_kwargs={"torch_dtype": torch.bfloat16},
     device_map="auto",
+    token=hf_key
 )
 
 # Define the function to generate text