Spaces:
Build error
Build error
Update app.py
Browse files
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
|