Update app.py

app.py

@@ -1,14 +1,14 @@
 from typing import Optional
 import uvicorn
 from fastapi import FastAPI, Form, File, UploadFile
-from transformers import AutoTokenizer, AutoModelForImageGeneration
+from transformers import AutoTokenizer, GPTNeoForCausalLM
 from PIL import Image
 import io
 
 app = FastAPI()
 
 tokenizer = AutoTokenizer.from_pretrained("EleutherAI/sd-1.5")
-model = AutoModelForImageGeneration.from_pretrained("EleutherAI/sd-1.5")
+model = GPTNeoForCausalLM.from_pretrained("EleutherAI/sd-1.5")
 
 @app.post("/generate_drawing/")
 async def generate_drawing(prompt: str = Form(...)):
@@ -19,4 +19,3 @@ async def generate_drawing(prompt: str = Form(...)):
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)
-