Update app.py

app.py

@@ -1,9 +1,9 @@
-import os
-os.system('pip install torch')
-
 import torch
 from transformers import pipeline
 import streamlit as st
+import os
+
+token = os.environ.get('HF_TOKEN')
 
 model_id = "google/gemma-2-2b-it"
 welcome_message = f"Hello there 👋! Is there anything I can help you with?"
@@ -15,6 +15,7 @@ def model_setup(model_id):
     model=model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
+    token=token,
     )
     return pipe