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