Hugging Face's logo

Spaces:
flax-community
/
code-clippy-problem-solver
Runtime error

App Files Community
arampacha commited on
Commit
c4d8cff
•
1 Parent(s): 6b904c9

upd model names

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import streamlit as st
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
- # model_name = "flax-community/gpt-code-clippy-1.3B-apps-alldata"
5
- model_name = "flax-community/gpt-code-clippy-125M-apps-alldata"
6
 
7
  @st.cache(allow_output_mutation=True, max_entries=1)
8
  def get_model():
 
1
  import streamlit as st
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
+ # model_name = "flax-community/gpt-neo-1.3B-apps-all"
5
+ model_name = "flax-community/gpt-neo-125M-apps-all"
6
 
7
  @st.cache(allow_output_mutation=True, max_entries=1)
8
  def get_model():