alexkueck commited on
Commit
96d0afc
1 Parent(s): 4383242

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -384,8 +384,9 @@ def invoke (prompt, file, history, rag_option, model_option, openai_api_key, k=3
384
  #LLM auswählen (OpenAI oder HF)
385
  ###########################
386
  if (model_option == "OpenAI"):
 
387
  #Anfrage an OpenAI ----------------------------
388
- if (prompt.value.find('Bild zeichnen')):
389
  llm = ChatOpenAI(model_name = MODEL_NAME_OAI_ZEICHNEN, openai_api_key = openai_api_key, temperature=temperature)#, top_p = top_p)
390
  else:
391
  llm = ChatOpenAI(model_name = MODEL_NAME, openai_api_key = openai_api_key, temperature=temperature)#, top_p = top_p)
 
384
  #LLM auswählen (OpenAI oder HF)
385
  ###########################
386
  if (model_option == "OpenAI"):
387
+ text = prompt
388
  #Anfrage an OpenAI ----------------------------
389
+ if (text.find('Bild zeichnen')):
390
  llm = ChatOpenAI(model_name = MODEL_NAME_OAI_ZEICHNEN, openai_api_key = openai_api_key, temperature=temperature)#, top_p = top_p)
391
  else:
392
  llm = ChatOpenAI(model_name = MODEL_NAME, openai_api_key = openai_api_key, temperature=temperature)#, top_p = top_p)