brdemorin commited on
Commit
a79eab8
1 Parent(s): 4a5c3aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ api_token = os.getenv("HF_API_TOKEN")
8
  if not api_token:
9
  raise ValueError("API token is missing. Please set it in Hugging Face Secrets.")
10
 
11
- # Replace with your model repository
12
- client = InferenceClient(model="brdemorin/Phi3_80_steps_v2", token=api_token)
13
 
14
  def respond(
15
  message,
 
8
  if not api_token:
9
  raise ValueError("API token is missing. Please set it in Hugging Face Secrets.")
10
 
11
+ # Replace with your model repository and specify the task
12
+ client = InferenceClient(model="brdemorin/Phi3_80_steps_v2", token=api_token, task="text-generation")
13
 
14
  def respond(
15
  message,