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

downgrade numpy

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,6 @@ import gradio as gr
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
10
  import spacy
11
  import subprocess
@@ -39,6 +37,8 @@ for package, version in packages.items():
39
  check_and_install(package, version)
40
 
41
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
42
 
43
  models = ["en_core_web_sm", "ru_core_news_sm", "de_core_news_sm",
44
  "ja_core_news_sm", "ko_core_news_sm", "es_core_news_sm"]
 
4
  import torch
5
  import random
6
  import logging
 
 
7
  from huggingface_hub import login, HfApi, snapshot_download
8
  import spacy
9
  import subprocess
 
37
  check_and_install(package, version)
38
 
39
  from transformers import AutoTokenizer, AutoModelForCausalLM
40
+ from vecalign.plan2align import translate_text, external_find_best_translation
41
+ from trl import AutoModelForCausalLMWithValueHead
42
 
43
  models = ["en_core_web_sm", "ru_core_news_sm", "de_core_news_sm",
44
  "ja_core_news_sm", "ko_core_news_sm", "es_core_news_sm"]