Spaces:
Paused
Paused
Commit
•
646eba0
1
Parent(s):
bdf0ce8
Update API base URL and remove unnecessary code
Browse files
main.py
CHANGED
@@ -183,7 +183,7 @@ async def predict(card: str, dataset_id: str) -> str | None:
|
|
183 |
try:
|
184 |
prompt = format_prompt(card)
|
185 |
client = AsyncOpenAI(
|
186 |
-
base_url="https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-v0.1/v1",
|
187 |
api_key=HF_TOKEN,
|
188 |
)
|
189 |
|
@@ -216,8 +216,7 @@ async def get_summary(dataset_id: str) -> str | None:
|
|
216 |
# dataset_id = request.dataset_id
|
217 |
try:
|
218 |
resp = await async_httpx_client.get(
|
219 |
-
f"https://huggingface.co/datasets/{dataset_id}/raw/main/README.md"
|
220 |
-
follow_redirects=True,
|
221 |
)
|
222 |
if resp.status_code != 200:
|
223 |
return None
|
|
|
183 |
try:
|
184 |
prompt = format_prompt(card)
|
185 |
client = AsyncOpenAI(
|
186 |
+
base_url="https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1/v1",
|
187 |
api_key=HF_TOKEN,
|
188 |
)
|
189 |
|
|
|
216 |
# dataset_id = request.dataset_id
|
217 |
try:
|
218 |
resp = await async_httpx_client.get(
|
219 |
+
f"https://huggingface.co/datasets/{dataset_id}/raw/main/README.md"
|
|
|
220 |
)
|
221 |
if resp.status_code != 200:
|
222 |
return None
|