Spaces:
Sleeping
Sleeping
Commit
·
ebf5cb5
1
Parent(s):
12eb652
Update app.py
Browse files
app.py
CHANGED
@@ -25,9 +25,9 @@ logging.basicConfig(
|
|
25 |
# adapter_model = "msuhail97/opt-1.3b-lora"
|
26 |
# tokenizer, model, device = load_tokenizer_and_model(base_model, adapter_model)
|
27 |
|
28 |
-
base_model = "facebook/opt-350m"
|
29 |
-
|
30 |
-
tokenizer, model, device = load_finetune_tokenizer_and_model(
|
31 |
|
32 |
|
33 |
total_count = 0
|
@@ -119,7 +119,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
|
|
119 |
with gr.Row():
|
120 |
gr.HTML(title)
|
121 |
status_display = gr.Markdown("Success", elem_id="status_display")
|
122 |
-
|
123 |
with gr.Row(scale=1).style(equal_height=True):
|
124 |
with gr.Column(scale=5):
|
125 |
with gr.Row(scale=1):
|
|
|
25 |
# adapter_model = "msuhail97/opt-1.3b-lora"
|
26 |
# tokenizer, model, device = load_tokenizer_and_model(base_model, adapter_model)
|
27 |
|
28 |
+
# base_model = "facebook/opt-350m"
|
29 |
+
finetune_model_path = "/ft_models/ft-opt-1.3b"
|
30 |
+
tokenizer, model, device = load_finetune_tokenizer_and_model(finetune_model_path)
|
31 |
|
32 |
|
33 |
total_count = 0
|
|
|
119 |
with gr.Row():
|
120 |
gr.HTML(title)
|
121 |
status_display = gr.Markdown("Success", elem_id="status_display")
|
122 |
+
|
123 |
with gr.Row(scale=1).style(equal_height=True):
|
124 |
with gr.Column(scale=5):
|
125 |
with gr.Row(scale=1):
|