Update chatbot_bedrock.py

chatbot_bedrock.py

@@ -7,14 +7,14 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import streamlit as st
 from langchain_community.llms import HuggingFaceHub
 
-
+my_model_id = os.getenv('MODEL_REPO_ID', 'Default Value')
 
 @st.cache_resource 
 def load_model():
     #tokenizer = AutoTokenizer.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b")
     #model = AutoModelForCausalLM.from_pretrained("KvrParaskevi/Hotel-Assistant-Attempt4-Llama-2-7b")
     model = HuggingFaceHub(
-        repo_id=MODEL_REPO_ID,
+        repo_id=my_model_id,
         task="text-generation",
         model_kwargs={
             "max_new_tokens": 512,