mike23415 commited on
Commit
580eaed
·
verified ·
1 Parent(s): 6166ea3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,10 +1,14 @@
1
  import os
 
2
  from pathlib import Path
3
  from flask import Flask, request, jsonify
4
  from flask_cors import CORS
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
  import torch
7
 
 
 
 
8
  # Create cache directory if not exists
9
  cache_dir = Path(os.getenv('TRANSFORMERS_CACHE', '/app/cache'))
10
  cache_dir.mkdir(parents=True, exist_ok=True)
 
1
  import os
2
+ import numpy as np
3
  from pathlib import Path
4
  from flask import Flask, request, jsonify
5
  from flask_cors import CORS
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
  import torch
8
 
9
+ # Verify numpy version
10
+ assert np.__version__.startswith('1.'), f"Invalid numpy version {np.__version__} - must be 1.x series"
11
+
12
  # Create cache directory if not exists
13
  cache_dir = Path(os.getenv('TRANSFORMERS_CACHE', '/app/cache'))
14
  cache_dir.mkdir(parents=True, exist_ok=True)