Update app.py

app.py

@@ -6,7 +6,7 @@ import time
 from langsmith import traceable
 import random
 
-### Begin Library Code
+
 from transformers import pipeline
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
@@ -25,10 +25,8 @@ tokenizer = None
 model = None
 model_name = "teapotai/teapotllm"
 with st.spinner('Loading Model...'):
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-
-
+    tokenizer = AutoTokenizer.from_pretrained(model_name, revision="699ab39cbf586674806354e92fbd6179f9a95f4a")
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name,revision="699ab39cbf586674806354e92fbd6179f9a95f4a")
 
 def log_time(func):
     def wrapper(*args, **kwargs):