kshahnathwani commited on
Commit
9dd37fa
·
verified ·
1 Parent(s): a12947f

update with gpt2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
61
 
62
  for chunk in client.text_generation(
63
  prompt,
64
- model="microsoft/Phi-3-mini-4k-instruct", # ✅ causal LM
65
  max_new_tokens=max_tokens,
66
  temperature=temperature,
67
  top_p=top_p,
 
61
 
62
  for chunk in client.text_generation(
63
  prompt,
64
+ model="gpt2", # ✅ causal LM
65
  max_new_tokens=max_tokens,
66
  temperature=temperature,
67
  top_p=top_p,