katsukiai commited on
Commit
d1155ef
·
verified ·
1 Parent(s): 8eb49e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ def log_error(error_msg):
42
  except Exception as e:
43
  logging.error(f"Failed to upload error log: {str(e)}")
44
 
45
- tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/deepseek-coder-6.7b-instruct")
46
- model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-coder-6.7b-instruct")
47
  meaning_generator = pipeline("text2text-generation", model="google/flan-t5-large")
48
 
49
  HF_TOKEN = os.getenv("HF_TOKEN", secrets.token_hex(16))
 
42
  except Exception as e:
43
  logging.error(f"Failed to upload error log: {str(e)}")
44
 
45
+ tokenizer = AutoTokenizer.from_pretrained("ai21labs/AI21-Jamba-Mini-1.6")
46
+ model = AutoModelForCausalLM.from_pretrained("ai21labs/AI21-Jamba-Mini-1.6")
47
  meaning_generator = pipeline("text2text-generation", model="google/flan-t5-large")
48
 
49
  HF_TOKEN = os.getenv("HF_TOKEN", secrets.token_hex(16))