awacke1 commited on
Commit
feabd29
·
verified ·
1 Parent(s): a947cb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -1,6 +1,7 @@
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 = []
@@ -33,10 +34,12 @@ def process_models(model_name, save_dir, additional_models):
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
- """
 
1
  import os
2
  import gradio as gr
3
  from transformers import AutoModel, AutoTokenizer
4
+ from fastapi import FastAPI
5
 
6
  def process_models(model_name, save_dir, additional_models):
7
  log_lines = []
 
34
 
35
  return "\n".join(log_lines)
36
 
37
+ # A Mermaid diagram embedded in Markdown as a mini-glossary.
38
+ mermaid_glossary = """
39
+ ```mermaid
40
+ graph LR
41
  A[🚀 Model Input] --> B[Load Model]
42
  B --> C[💾 Save Model]
43
  D[🧩 Additional Models] --> B
44
+ ```
45
+ """