w831152001 commited on
Commit
a58ba2a
·
1 Parent(s): 36a7566
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. main.py +1 -1
Dockerfile CHANGED
@@ -8,4 +8,4 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
9
  COPY . .
10
 
11
- CMD ["gunicorn","-b", "0.0.0.0:7860", "main:app"]
 
8
 
9
  COPY . .
10
 
11
+ CMD ["gunicorn","-b", "0.0.0.0:7860", "main:app"]
main.py CHANGED
@@ -32,7 +32,7 @@ API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-In
32
  def query(payload):
33
  response = requests.post(API_URL, headers=headers, json=payload)
34
  app.logger.info("-----"+response.text)
35
- return response
36
 
37
  app = Flask(__name__)
38
 
 
32
  def query(payload):
33
  response = requests.post(API_URL, headers=headers, json=payload)
34
  app.logger.info("-----"+response.text)
35
+ return response.text
36
 
37
  app = Flask(__name__)
38