Mahmoud Amiri
commited on
Commit
·
62a8987
1
Parent(s):
8aa76ff
add temp cell to print version numberr
Browse files
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 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|