Update app.py

app.py

@@ -45,10 +45,7 @@ def visualize_diagram(input_text):
     print (mermaid_code)
     url = generate_diagram(mermaid_code[0])
     print ("Url",url)
-    #img = Image.open(requests.get(url, stream=True).raw)
-    client = httpx.Client()
-    response = client.get(url)
-    img = Image.open(response.content)
+    img = Image.open(requests.get(url, headers={'content-type': 'image/png'},stream=True).raw)
     print ("Image",img)
     return img