Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ def generate_description(caption):
|
|
26 |
|
27 |
prompt = PromptTemplate(template=template, input_variables=["caption"])
|
28 |
|
29 |
-
desc_llm = LLMChain(llm=OpenAI(model_name="gpt-4", temperature=1), prompt=prompt, verbose=True)
|
30 |
description = desc_llm.predict(caption=caption).replace('"', '')
|
31 |
|
32 |
print(description)
|
|
|
26 |
|
27 |
prompt = PromptTemplate(template=template, input_variables=["caption"])
|
28 |
|
29 |
+
desc_llm = LLMChain(llm=OpenAI(model_name="gpt-4-1106-preview", temperature=1), prompt=prompt, verbose=True)
|
30 |
description = desc_llm.predict(caption=caption).replace('"', '')
|
31 |
|
32 |
print(description)
|