Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ if not os.path.exists("./models"):
|
|
28 |
|
29 |
hf_hub_download(
|
30 |
repo_id="bartowski/google_gemma-3-1b-it-GGUF",
|
31 |
-
filename="google_gemma-3-1b-it-
|
32 |
local_dir="./models",
|
33 |
)
|
34 |
hf_hub_download(
|
@@ -73,7 +73,7 @@ llm_model = None
|
|
73 |
def respond(
|
74 |
message: str,
|
75 |
history: List[Tuple[str, str]],
|
76 |
-
model: str = "google_gemma-3-1b-it-
|
77 |
system_message: str = "You are a helpful assistant.",
|
78 |
max_tokens: int = 1024,
|
79 |
temperature: float = 0.7,
|
@@ -105,7 +105,7 @@ def respond(
|
|
105 |
|
106 |
# Ensure model is not None
|
107 |
if model is None:
|
108 |
-
model = "google_gemma-3-1b-it-
|
109 |
|
110 |
# Load the model
|
111 |
if llm is None or llm_model != model:
|
@@ -187,10 +187,10 @@ demo = gr.ChatInterface(
|
|
187 |
additional_inputs=[
|
188 |
gr.Dropdown(
|
189 |
choices=[
|
190 |
-
"google_gemma-3-1b-it-
|
191 |
"google_gemma-3-1b-it-Q5_K_M.gguf",
|
192 |
],
|
193 |
-
value="google_gemma-3-1b-it-
|
194 |
label="Model",
|
195 |
info="Select the AI model to use for chat",
|
196 |
),
|
|
|
28 |
|
29 |
hf_hub_download(
|
30 |
repo_id="bartowski/google_gemma-3-1b-it-GGUF",
|
31 |
+
filename="google_gemma-3-1b-it-Q4_K_M.gguf",
|
32 |
local_dir="./models",
|
33 |
)
|
34 |
hf_hub_download(
|
|
|
73 |
def respond(
|
74 |
message: str,
|
75 |
history: List[Tuple[str, str]],
|
76 |
+
model: str = "google_gemma-3-1b-it-Q4_K_M.gguf", # Set default model
|
77 |
system_message: str = "You are a helpful assistant.",
|
78 |
max_tokens: int = 1024,
|
79 |
temperature: float = 0.7,
|
|
|
105 |
|
106 |
# Ensure model is not None
|
107 |
if model is None:
|
108 |
+
model = "google_gemma-3-1b-it-Q4_K_M.gguf"
|
109 |
|
110 |
# Load the model
|
111 |
if llm is None or llm_model != model:
|
|
|
187 |
additional_inputs=[
|
188 |
gr.Dropdown(
|
189 |
choices=[
|
190 |
+
"google_gemma-3-1b-it-Q4_K_M.gguf",
|
191 |
"google_gemma-3-1b-it-Q5_K_M.gguf",
|
192 |
],
|
193 |
+
value="google_gemma-3-1b-it-Q4_K_M.gguf",
|
194 |
label="Model",
|
195 |
info="Select the AI model to use for chat",
|
196 |
),
|