sitammeur commited on
Commit
42f7208
·
verified ·
1 Parent(s): deedc69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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-Q6_K.gguf",
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-Q5_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,7 +105,7 @@ def respond(
105
 
106
  # Ensure model is not None
107
  if model is None:
108
- model = "google_gemma-3-1b-it-Q5_K_M.gguf"
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-Q6_K.gguf",
191
  "google_gemma-3-1b-it-Q5_K_M.gguf",
192
  ],
193
- value="google_gemma-3-1b-it-Q5_K_M.gguf",
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
  ),