Update app.py

app.py

@@ -351,7 +351,7 @@ def main():
                                         
                             class_name = trainclass
                                         
-                            query_text ="a "+prompt.split()[1]+" of a "+class_name
+                            query_text ="a photograph of a "+class_name
                             c_split = model.cond_stage_model.tokenizer.tokenize(query_text)
                                         
                             sen_text_embedding = model.get_learned_conditioning(query_text)