shee2205 commited on
Commit
295b198
·
verified ·
1 Parent(s): 272f768

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def build_context(profile, daily):
34
  def get_llm():
35
  return pipeline(
36
  "text-generation",
37
- model="google/flan-t5-small",
38
  max_new_tokens=128,
39
  do_sample=True,
40
  temperature=0.7,
 
34
  def get_llm():
35
  return pipeline(
36
  "text-generation",
37
+ model="gpt2",
38
  max_new_tokens=128,
39
  do_sample=True,
40
  temperature=0.7,