red1xe commited on
Commit
fc6f1e0
1 Parent(s): 56e7c58
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ login(token='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', add_to_git_credential=True)
14
 
15
  st.title("Code Generation")
16
  st.write('MODEL: TinyPixel/red1xe/Llama-2-7B-codeGPT')
17
- model_name='red1xe/Llama-2-7B-codeGPT'
18
  tokenizer = AutoTokenizer.from_pretrained(model_name)
19
  model= AutoModelForCausalLM.from_pretrained(model_name)
20
 
 
14
 
15
  st.title("Code Generation")
16
  st.write('MODEL: TinyPixel/red1xe/Llama-2-7B-codeGPT')
17
+ model_name='lmsys/vicuna-7b-v1.1'
18
  tokenizer = AutoTokenizer.from_pretrained(model_name)
19
  model= AutoModelForCausalLM.from_pretrained(model_name)
20