KuangDW commited on
Commit
3c0f52c
·
1 Parent(s): c4834c2

downgrade numpy

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -4,8 +4,6 @@ import gradio as gr
4
  import torch
5
  import random
6
  import logging
7
- import openai
8
- from openai import OpenAI
9
  from vecalign.plan2align import translate_text, external_find_best_translation
10
  from trl import AutoModelForCausalLMWithValueHead
11
  from huggingface_hub import login, HfApi, snapshot_download
@@ -17,9 +15,6 @@ import sys
17
  login(token=os.environ.get("LA_NAME"))
18
  os.environ["LASER"] = "laser"
19
 
20
- import transformers
21
- print(transformers.__version__)
22
-
23
  def check_and_install(package, required_version):
24
  try:
25
  dist = pkg_resources.get_distribution(package)
@@ -34,11 +29,12 @@ def check_and_install(package, required_version):
34
  subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}=={required_version}"])
35
  packages = {
36
  "pip": "24.0",
 
37
  "fairseq": "0.12.2",
38
  "torch": "2.6.0",
39
- "transformers": "4.51.3",
40
- "numpy": "1.24.0"
41
  }
 
42
  for package, version in packages.items():
43
  check_and_install(package, version)
44
 
 
4
  import torch
5
  import random
6
  import logging
 
 
7
  from vecalign.plan2align import translate_text, external_find_best_translation
8
  from trl import AutoModelForCausalLMWithValueHead
9
  from huggingface_hub import login, HfApi, snapshot_download
 
15
  login(token=os.environ.get("LA_NAME"))
16
  os.environ["LASER"] = "laser"
17
 
 
 
 
18
  def check_and_install(package, required_version):
19
  try:
20
  dist = pkg_resources.get_distribution(package)
 
29
  subprocess.check_call([sys.executable, "-m", "pip", "install", f"{package}=={required_version}"])
30
  packages = {
31
  "pip": "24.0",
32
+ "numpy": "1.24.0",
33
  "fairseq": "0.12.2",
34
  "torch": "2.6.0",
35
+ "transformers": "4.51.3"
 
36
  }
37
+
38
  for package, version in packages.items():
39
  check_and_install(package, version)
40