Hugging Face's logo

Spaces:
joshuaberkowitzus
/
txgemma-2b-predict-demo
Runtime error

App Files Community
joshuaberkowitzus commited on
Commit
4360f38
·
verified ·
1 Parent(s): c71a68d

back to 2B

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,8 +13,8 @@ import os # Import os library for path joining
13
 
14
 
15
  # --- Configuration ---
16
- #MODEL_NAME = "google/txgemma-2b-predict"
17
- MODEL_NAME = "google/txgemma-9b-predict"
18
  PROMPT_FILENAME = "tdc_prompts.json"
19
  MODEL_CACHE = "model_cache" # Optional: define a cache directory
20
  # MAX_EXAMPLES is no longer strictly limiting the display, but can be used if needed later
 
13
 
14
 
15
  # --- Configuration ---
16
+ MODEL_NAME = "google/txgemma-2b-predict"
17
+ #MODEL_NAME = "google/txgemma-9b-predict"
18
  PROMPT_FILENAME = "tdc_prompts.json"
19
  MODEL_CACHE = "model_cache" # Optional: define a cache directory
20
  # MAX_EXAMPLES is no longer strictly limiting the display, but can be used if needed later