Update app.py

app.py

@@ -13,7 +13,7 @@ def get_model(name_of_model):
     model = AutoModelForCausalLM.from_pretrained(name_of_model)
     return model, tokenizer
     
-model, tokenizer = get_model()
+model, tokenizer = get_model(name_of_model)
 temp = st.sidebar.slider("Temperature", 0.7, 1.5)
 number_of_outputs = st.sidebar.slider("Number of Outputs", 5, 50)
 lengths = st.sidebar.slider("Length", 3, 500)