PhantHive commited on
Commit
4cf26ce
·
verified ·
1 Parent(s): 60093a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,7 +9,6 @@ model_id = "phearion/bigbrain-v0.0.1"
9
 
10
  # Load models and tokenizer efficiently
11
  config = PeftConfig.from_pretrained("phearion/bigbrain-v0.0.1")
12
-
13
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
14
  model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path)
15
 
@@ -24,4 +23,4 @@ def greet(text):
24
  return tokenizer.decode(output_tokens[0], skip_special_tokens=True)
25
 
26
  iface = gr.Interface(fn=greet, inputs="text", outputs="text", title="PEFT Model for Big Brain")
27
- iface.launch(share=True) # Share directly to Gradio Space
 
9
 
10
  # Load models and tokenizer efficiently
11
  config = PeftConfig.from_pretrained("phearion/bigbrain-v0.0.1")
 
12
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
13
  model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path)
14
 
 
23
  return tokenizer.decode(output_tokens[0], skip_special_tokens=True)
24
 
25
  iface = gr.Interface(fn=greet, inputs="text", outputs="text", title="PEFT Model for Big Brain")
26
+ iface.launch() # Share directly to Gradio Space