ybelkada commited on
Commit
9031ab2
·
verified ·
1 Parent(s): a452ab2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -28,14 +28,12 @@ def script_to_use(model_id, api):
28
  arch = arch[0]
29
  return "convert.py" if arch in LLAMA_LIKE_ARCHS else "convert-hf-to-gguf.py"
30
 
31
- def process_model(model_id, q_method, private_repo, oauth_token: gr.OAuthToken | None):
32
- if oauth_token.token is None:
33
- raise ValueError("You must be logged in to use GGUF-my-repo")
34
  model_name = model_id.split('/')[-1]
35
  fp16 = f"{model_name}/{model_name.lower()}.fp16.bin"
36
 
37
  try:
38
- api = HfApi(token=oauth_token.token)
39
 
40
  dl_pattern = ["*.md", "*.json", "*.model"]
41
 
@@ -77,7 +75,7 @@ def process_model(model_id, q_method, private_repo, oauth_token: gr.OAuthToken |
77
  print("Repo created successfully!", new_repo_url)
78
 
79
  try:
80
- card = ModelCard.load(model_id, token=oauth_token.token)
81
  except:
82
  card = ModelCard("")
83
  if card.data.tags is None:
@@ -167,6 +165,10 @@ iface = gr.Interface(
167
  label="Private Repo",
168
  info="Create a private repo under your username."
169
  ),
 
 
 
 
170
  ],
171
  outputs=[
172
  gr.Markdown(label="output"),
 
28
  arch = arch[0]
29
  return "convert.py" if arch in LLAMA_LIKE_ARCHS else "convert-hf-to-gguf.py"
30
 
31
+ def process_model(model_id, q_method, private_repo, token: str | None):
 
 
32
  model_name = model_id.split('/')[-1]
33
  fp16 = f"{model_name}/{model_name.lower()}.fp16.bin"
34
 
35
  try:
36
+ api = HfApi(token=token)
37
 
38
  dl_pattern = ["*.md", "*.json", "*.model"]
39
 
 
75
  print("Repo created successfully!", new_repo_url)
76
 
77
  try:
78
+ card = ModelCard.load(model_id, token=token)
79
  except:
80
  card = ModelCard("")
81
  if card.data.tags is None:
 
165
  label="Private Repo",
166
  info="Create a private repo under your username."
167
  ),
168
+ gr.Text(
169
+ label="HF Token",
170
+ info="Your HF Token."
171
+ ),
172
  ],
173
  outputs=[
174
  gr.Markdown(label="output"),