Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
66ef2ad
1
Parent(s):
3f5a139
no more env
Browse files- .env +0 -1
- 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(
|