Update app.py

app.py

@@ -1,14 +1,24 @@
-import streamlit as st
-from transformers import AutoTokenizer
-model_name ="NousResearch/Llama-2-7b-chat-hf"
-tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
-tokenizer.pad_token = tokenizer.eos_token
-tokenizer.padding_side = "right" 
-
-
-model = "Hardik1234/llama-finetune-reactjs"
-pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=2048)
-prompt = st.text_area('Enter prompt: ')
-if prompt:
-    result = pipe(f"<s> [INST] {prompt} [/INST] ")
-    st.json(result[0]['generated_text'])
+try:
+    
+        
+    import streamlit as st
+    from transformers import AutoTokenizer
+    model_name ="NousResearch/Llama-2-7b-chat-hf"
+    print('tokenizer_loading')
+    tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+    tokenizer.pad_token = tokenizer.eos_token
+    tokenizer.padding_side = "right" 
+    print('tokenizer_loaded')
+    
+    model = "Hardik1234/llama-finetune-reactjs"
+    print('loading_model')
+    pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=2048)
+    print('model_loaded')
+    prompt = st.text_area('Enter prompt: ')
+    if prompt:
+        print('taking prompt')
+        result = pipe(f"<s> [INST] {prompt} [/INST] ")
+        print('generating output')
+        st.json(result[0]['generated_text'])
+except error:
+    print(error)