dsuyu1 commited on
Commit
2483751
·
verified ·
1 Parent(s): 138b76e

Fixed auth issues (no longer gated)

Browse files

removed required auth token explicitly to make the model public, also refreshed cache of model hopefully that fixes something

Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -1,3 +1,9 @@
 
 
 
 
 
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
@@ -10,6 +16,7 @@ model = AutoModelForCausalLM.from_pretrained(
10
  torch_dtype=torch.float16,
11
  device_map="auto",
12
  load_in_8bit=True
 
13
  )
14
 
15
  def generate_playbook(incident_type, target_asset, detection_source,
 
1
+ from huggingface_hub import HfApi
2
+ import os
3
+
4
+ # Clear the cache for your model
5
+ os.system("rm -rf ~/.cache/huggingface/hub/models--dsuyu1--rascal")
6
+
7
  import gradio as gr
8
  from transformers import AutoModelForCausalLM, AutoTokenizer
9
  import torch
 
16
  torch_dtype=torch.float16,
17
  device_map="auto",
18
  load_in_8bit=True
19
+ use_auth_token=False
20
  )
21
 
22
  def generate_playbook(incident_type, target_asset, detection_source,