holytinz278 commited on
Commit
ce5bdbb
·
verified ·
1 Parent(s): dc57d74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
  # Initialize the client with the fine-tuned model
5
- client = InferenceClient("albert/albert-base-v2") # Update if using another model
6
 
7
  # Model's token limit
8
  MODEL_TOKEN_LIMIT = 16384
 
2
  from huggingface_hub import InferenceClient
3
 
4
  # Initialize the client with the fine-tuned model
5
+ client = InferenceClient("deepseek-ai/DeepSeek-R1") # Update if using another model
6
 
7
  # Model's token limit
8
  MODEL_TOKEN_LIMIT = 16384