Update app.py
Browse files
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(
|
|
|
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
|