Update app.py

app.py

@@ -11,14 +11,14 @@ import urllib.request
 import uuid
 import requests
 import io
-import chat_models
+from chat_models import models as c_models
 
 loaded_model=[]
 chat_model=[]
 for i,model in enumerate(models):
     loaded_model.append(gr.load(f'models/{model}'))
 print (loaded_model)
-for i,model_c in enumerate(chat_models.models):
+for i,model_c in enumerate(c_models):
     chat_model.append(model_c)
 print (chat_model)
 now = datetime.now()
@@ -58,20 +58,20 @@ def run_gpt(in_prompt,history,model_drop):
     return resp
 
 
-def generate(purpose,history,model_drop,chat_drop):
+def generate(purpose,history,chat_drop):
     print (history)
-    out_prompt = run_gpt(purpose,history,model_drop)
+    out_prompt = run_gpt(purpose,history,chat_drop)
     return out_prompt
 
-def describe(purpose,history,model_drop,chat_drop):
+def describe(purpose,history,chat_drop):
     print (history)
-    out_prompt = run_gpt(purpose,history,model_drop)
+    out_prompt = run_gpt(purpose,history,chat_drop)
     return out_prompt
    
 def run(purpose,history,model_drop,chat_drop,choice):
     if choice == "Generate":
         #out_img = infer(out_prompt)
-        out_prompt=generate(purpose,history,model_drop,chat_drop)
+        out_prompt=generate(purpose,history,chat_drop)
         history.append((purpose,out_prompt))
         yield (history,None)        
         model=loaded_model[int(model_drop)]