Rahatara commited on
Commit
a32087c
·
verified ·
1 Parent(s): cd4ec48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,8 +57,8 @@ def generate_variations(base_prompt, number_of_variations):
57
  weather = random.choice(weather_conditions)
58
 
59
  # Enhance the base prompt with the GPT model
60
- enhanced_prompt = ask_rail_defect_question(base_prompt)
61
-
62
  full_prompt = f"{enhanced_prompt}, with a {size} defect {location}, observed {weather}."
63
  variations.append(full_prompt)
64
  return variations
 
57
  weather = random.choice(weather_conditions)
58
 
59
  # Enhance the base prompt with the GPT model
60
+ #enhanced_prompt = ask_rail_defect_question(base_prompt)
61
+ enhanced_prompt = base_prompt
62
  full_prompt = f"{enhanced_prompt}, with a {size} defect {location}, observed {weather}."
63
  variations.append(full_prompt)
64
  return variations