debug...

.gitignore

@@ -1,2 +1,3 @@
 # environment
 bloom_demo
+tutorial-env

main.py

@@ -42,17 +42,19 @@ def model_init():
 
 tokenizer, model = model_init()
 
-# ===================== INPUT ====================== #
-# prompt = "\u554F\uFF1A\u53F0\u7063\u6700\u9AD8\u7684\u5EFA\u7BC9\u7269\u662F\uFF1F\u7B54\uFF1A"  #@param {type:"string"}
-prompt = st.text_input("Prompt: ")
+try:
+    # ===================== INPUT ====================== #
+    # prompt = "\u554F\uFF1A\u53F0\u7063\u6700\u9AD8\u7684\u5EFA\u7BC9\u7269\u662F\uFF1F\u7B54\uFF1A"  #@param {type:"string"}
+    prompt = st.text_input("Prompt: ")
 
-# =================== INFERENCE ==================== #
-if prompt:
-    with torch.no_grad():
-        [texts_out] = model.generate(
-            **tokenizer(
-                prompt, return_tensors="pt"
-            ).to(device))
-    output_text = tokenizer.decode(texts_out)
-    st.markdown(output_text)
-    
+    # =================== INFERENCE ==================== #
+    if prompt:
+        with torch.no_grad():
+            [texts_out] = model.generate(
+                **tokenizer(
+                    prompt, return_tensors="pt"
+                ).to(device))
+        output_text = tokenizer.decode(texts_out)
+        st.markdown(output_text)
+except Exception as err:
+    st.write(str(err))