attilasimko commited on
Commit
66ef2ad
·
1 Parent(s): 3f5a139

no more env

Browse files
Files changed (2) hide show
  1. .env +0 -1
  2. evaluations/models.py +1 -5
.env DELETED
@@ -1 +0,0 @@
1
- githubToken="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"
 
 
evaluations/models.py CHANGED
@@ -25,11 +25,7 @@ class LocalLLM():
25
  class RemoteLLM():
26
  def __init__(self, model_name):
27
  token = os.getenv("hfToken")
28
- self.client = InferenceClient(
29
- model_name,
30
- token=token,
31
- )
32
-
33
 
34
  def predict(self, response_type, prompt):
35
  for message in self.client.chat_completion(
 
25
  class RemoteLLM():
26
  def __init__(self, model_name):
27
  token = os.getenv("hfToken")
28
+ self.client = InferenceClient(model_name, token=token)
 
 
 
 
29
 
30
  def predict(self, response_type, prompt):
31
  for message in self.client.chat_completion(