Update app.py

app.py

@@ -17,7 +17,7 @@ now = datetime.now()
 date_time_str = now.strftime("%Y-%m-%d %H:%M:%S")
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
-model = gr.load("models/stabilityai/sdxl-turbo")
+#model = gr.load("models/stabilityai/sdxl-turbo")
 history = []
 
 def infer(txt):
@@ -37,7 +37,7 @@ def run_gpt(in_prompt,history):
     print (seed)
     generate_kwargs = dict(
         temperature=1.0,
-        max_new_tokens=128,
+        max_new_tokens=256,
         top_p=0.99,
         repetition_penalty=1.0,
         do_sample=True,
@@ -65,8 +65,8 @@ def run(purpose,history,model_drop):
         out_prompt = f"An Error Occured generating the prompt \n {e}"
     yield ("",[(purpose,out_prompt)],None)
     try:
-        #model=loaded_model[int(model_drop)]
-        out_img=model(f'cat')
+        model=loaded_model[int(model_drop)]
+        out_img=model(out_prompt)
         print(out_img)
         image=f'{base_url}file={out_img}'
         uid = uuid.uuid4()