davanstrien HF staff commited on
Commit
3d8a92f
·
1 Parent(s): 8ae1055

change weight

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -109,14 +109,14 @@ def weighted_random_choice(choices: List[Tuple[Callable, float]]) -> Callable:
109
 
110
  def generate_prompt() -> str:
111
  prompt_choices = [
112
- (detailed_genre_description_prompt, 0.4),
113
  (basic_prompt, 0.3),
114
- (very_basic_prompt, 0.3),
115
  ]
116
 
117
  selected_prompt_func = weighted_random_choice(prompt_choices)
118
  return selected_prompt_func()
119
-
120
  def get_and_store_prompt():
121
  prompt = generate_prompt()
122
  print(prompt) # Keep this for debugging
 
109
 
110
  def generate_prompt() -> str:
111
  prompt_choices = [
112
+ (detailed_genre_description_prompt, 0.5),
113
  (basic_prompt, 0.3),
114
+ (very_basic_prompt, 0.2),
115
  ]
116
 
117
  selected_prompt_func = weighted_random_choice(prompt_choices)
118
  return selected_prompt_func()
119
+
120
  def get_and_store_prompt():
121
  prompt = generate_prompt()
122
  print(prompt) # Keep this for debugging