davanstrien HF staff commited on
Commit
646eba0
1 Parent(s): bdf0ce8

Update API base URL and remove unnecessary code

Browse files
Files changed (1) hide show
  1. main.py +2 -3
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