santoshNA commited on
Commit
9551109
·
1 Parent(s): 8666b71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)