Update app.py

app.py

@@ -4,7 +4,7 @@ from PIL import Image
 import gradio as gr
 
 # Load text-to-image model
-text_to_image_model = torch.hub.load("ProPerNounpYK/texttoimage", map_location=torch.device("cuda" if torch.cuda.is_available() else "cpu"))
+text_to_image_model = torch.hub.load("ProPerNounpYK/texttoimage", "generator", map_location=torch.device("cuda" if torch.cuda.is_available() else "cpu"))
 
 # Load chat model
 chat_model = AutoModelForSeq2SeqLM.from_pretrained("ProPerNounpYK/chat")