drmasad commited on
Commit
b431bd6
·
verified ·
1 Parent(s): 8dbaa52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -114,7 +114,7 @@ if prompt := st.chat_input("Ask me anything about diabetes"):
114
  with st.chat_message("assistant"):
115
  result = pipeline(
116
  task="text-generation",
117
- model=model,
118
  tokenizer=tokenizer,
119
  max_length=1024,
120
  temperature=temp_values
 
114
  with st.chat_message("assistant"):
115
  result = pipeline(
116
  task="text-generation",
117
+ model=model_name,
118
  tokenizer=tokenizer,
119
  max_length=1024,
120
  temperature=temp_values