Update app.py

app.py

@@ -16,6 +16,10 @@ from torch import autocast
 
 openai.api_key = os.getenv('openaikey')
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = StableDiffusionPipeline.from_pretrained("stale2000/sd-dnditem", torch_dtype=torch.float16)  
+pipe = pipe.to(device)
+
 def predict(input, manual_query_repacement, history=[]):
 
     # gpt3
@@ -87,12 +91,3 @@ gr.Interface(fn=predict,
              inputs=[inputText,manual_query,'state'],
             
              outputs=["chatbot",'state', output_img]).launch()
-
-
-
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = StableDiffusionPipeline.from_pretrained("stale2000/sd-dnditem", torch_dtype=torch.float16)  
-pipe = pipe.to(device)
-
-
-