Mahmoud Amiri commited on
Commit
62a8987
·
1 Parent(s): 8aa76ff

add temp cell to print version numberr

Browse files
Files changed (1) hide show
  1. app.py +18 -8
app.py CHANGED
@@ -2,6 +2,24 @@ import torch
2
  import gradio as gr
3
  from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer, AutoConfig
4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  # Model name
6
  model_name = "Bocklitz-Lab/lit2vec-tldr-bart-model"
7
 
@@ -63,12 +81,4 @@ with gr.Blocks() as demo:
63
  outputs=output_text)
64
 
65
  demo.launch()
66
- import pkg_resources
67
 
68
- # List only specific packages (e.g., gradio, transformers, etc.)
69
- for package_name in ["gradio", "transformers", "torch", "sentencepiece"]:
70
- try:
71
- version = pkg_resources.get_distribution(package_name).version
72
- print(f"{package_name}=={version}")
73
- except pkg_resources.DistributionNotFound:
74
- print(f"{package_name} not installed")
 
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
24
  model_name = "Bocklitz-Lab/lit2vec-tldr-bart-model"
25
 
 
81
  outputs=output_text)
82
 
83
  demo.launch()
 
84