awacke1 commited on
Commit
f451bf7
Β·
verified Β·
1 Parent(s): 5078e54

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +42 -0
app.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import gradio as gr
3
+ from transformers import AutoModel, AutoTokenizer
4
+
5
+ def process_models(model_name, save_dir, additional_models):
6
+ log_lines = []
7
+
8
+ # Process primary model
9
+ log_lines.append(f"πŸš€ Loading model: **{model_name}**")
10
+ try:
11
+ model = AutoModel.from_pretrained(model_name)
12
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
13
+ model_save_path = os.path.join(save_dir, model_name.replace("/", "_"))
14
+ os.makedirs(model_save_path, exist_ok=True)
15
+ model.save_pretrained(model_save_path)
16
+ log_lines.append(f"βœ… Saved **{model_name}** to `{model_save_path}`")
17
+ except Exception as e:
18
+ log_lines.append(f"❌ Error with **{model_name}**: {e}")
19
+
20
+ # Process additional models if any
21
+ if additional_models:
22
+ for m in additional_models:
23
+ log_lines.append(f"πŸš€ Loading model: **{m}**")
24
+ try:
25
+ model = AutoModel.from_pretrained(m)
26
+ tokenizer = AutoTokenizer.from_pretrained(m)
27
+ model_save_path = os.path.join(save_dir, m.replace("/", "_"))
28
+ os.makedirs(model_save_path, exist_ok=True)
29
+ model.save_pretrained(model_save_path)
30
+ log_lines.append(f"βœ… Saved **{m}** to `{model_save_path}`")
31
+ except Exception as e:
32
+ log_lines.append(f"❌ Error with **{m}**: {e}")
33
+
34
+ return "\n".join(log_lines)
35
+
36
+ # Mermaid glossary: a one-line flow summary of our UI actions.
37
+ mermaid_glossary = """graph LR
38
+ A[πŸš€ Model Input] --> B[Load Model]
39
+ B --> C[πŸ’Ύ Save Model]
40
+ D[🧩 Additional Models] --> B
41
+
42
+ """