Mahmoud Amiri commited on
Commit
5df42b4
·
1 Parent(s): 62a8987

fixed the requirenment file

Browse files
Files changed (2) hide show
  1. app.py +10 -16
  2. requirements.txt +3 -3
app.py CHANGED
@@ -2,22 +2,16 @@ import torch
2
  import gradio as gr
3
  from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoConfig
4
 
5
- import torch
6
- import gradio as gr
7
- from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoConfig
8
- import sys
9
- import pkg_resources
10
-
11
- # Log current package versions
12
- for package_name in ["gradio", "transformers", "torch", "sentencepiece"]:
13
- try:
14
- version = pkg_resources.get_distribution(package_name).version
15
- print(f"{package_name}=={version}", file=sys.stderr)
16
- except pkg_resources.DistributionNotFound:
17
- print(f"{package_name} not installed", file=sys.stderr)
18
-
19
-
20
-
21
 
22
 
23
  # Model name
 
2
  import gradio as gr
3
  from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoConfig
4
 
5
+ # import sys
6
+ # import pkg_resources
7
+
8
+ # # Log current package versions
9
+ # for package_name in ["gradio", "transformers", "torch", "sentencepiece"]:
10
+ # try:
11
+ # version = pkg_resources.get_distribution(package_name).version
12
+ # print(f"{package_name}=={version}", file=sys.stderr)
13
+ # except pkg_resources.DistributionNotFound:
14
+ # print(f"{package_name} not installed", file=sys.stderr)
 
 
 
 
 
 
15
 
16
 
17
  # Model name
requirements.txt CHANGED
@@ -1,3 +1,3 @@
1
- transformers
2
- torch
3
- gradio
 
1
+ gradio==5.42.0
2
+ transformers==4.55.3
3
+ torch==2.8.0