daydayup1225 commited on
Commit
ebf5cb5
·
1 Parent(s): 12eb652

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- # finetune_model_path = "/home2/xqchang/Chat-web/ft_models/ft-opt-1.3b"
30
- tokenizer, model, device = load_finetune_tokenizer_and_model(base_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
- gr.Markdown(description_top)
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):