FrameRateTech commited on
Commit
addeff3
·
verified ·
1 Parent(s): eaa0178

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ from transformers import (
10
  # ---------------------------------------------------------------------
11
  # 1. Model Configuration
12
  # ---------------------------------------------------------------------
13
- ADAPTER_ID = "FrameRateTech/DamageScan-llama-8b-instruct"
14
 
15
  # Load adapter config to find base model name
16
  peft_config = PeftConfig.from_pretrained(ADAPTER_ID)
@@ -111,7 +111,7 @@ def predict(messages, temperature, top_p, max_new_tokens):
111
  # 7. Build the Gradio Interface
112
  # ---------------------------------------------------------------------
113
  with gr.Blocks() as demo:
114
- gr.Markdown("<h1 align='center'>FrameRateTech/DamageScan-llama-8b-instruct Chatbot</h1>")
115
 
116
  with gr.Row():
117
  with gr.Column():
 
10
  # ---------------------------------------------------------------------
11
  # 1. Model Configuration
12
  # ---------------------------------------------------------------------
13
+ ADAPTER_ID = "FrameRateTech/DamageScan-llama-8b-instruct-merged"
14
 
15
  # Load adapter config to find base model name
16
  peft_config = PeftConfig.from_pretrained(ADAPTER_ID)
 
111
  # 7. Build the Gradio Interface
112
  # ---------------------------------------------------------------------
113
  with gr.Blocks() as demo:
114
+ gr.Markdown("<h1 align='center'>FrameRateTech/DamageScan-llama-8b-instruct-merged Chatbot</h1>")
115
 
116
  with gr.Row():
117
  with gr.Column():