Rahul-8799 commited on
Commit
33ecdc5
·
verified ·
1 Parent(s): ae75831

Update utils/inference.py

Browse files
Files changed (1) hide show
  1. utils/inference.py +6 -1
utils/inference.py CHANGED
@@ -1,6 +1,11 @@
1
  from huggingface_hub import InferenceClient
 
2
 
3
- client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
 
 
 
 
4
 
5
  def call_model(prompt: str) -> str:
6
  return client.text_generation(prompt, max_new_tokens=2048, temperature=0.3, return_full_text=False)
 
1
  from huggingface_hub import InferenceClient
2
+ import os
3
 
4
+ client = InferenceClient(
5
+ model="mistralai/Mistral-7B-Instruct-v0.2",
6
+ token=os.environ.get("HF_TOKEN"),
7
+ provider="hf-inference"
8
+ )
9
 
10
  def call_model(prompt: str) -> str:
11
  return client.text_generation(prompt, max_new_tokens=2048, temperature=0.3, return_full_text=False)