json

Dockerfile

@@ -8,4 +8,4 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
-CMD ["gunicorn","-b", "0.0.0.0:7860", "main:app"]
+CMD ["gunicorn","-b", "0.0.0.0:7860", "main:app"]

main.py

@@ -24,14 +24,13 @@ from linebot.v3.webhooks import (
 import os
 import requests
 import logging
-import json
 
 HF_TOKEN = os.environ.get('HF_TOKEN')
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
-    return json.dumps(response.text)
+    return response.text
 
 app = Flask(__name__)