Upload handler.py with huggingface_hub

handler.py

@@ -36,8 +36,7 @@ class EndpointHandler:
             prompt = prompts[0] if prompts else ""
         
         # Generate a placeholder SVG
-        model_name = os.path.basename(os.getcwd())
-        svg = f'<svg xmlns="http://www.w3.org/2000/svg" width="512" height="512" viewBox="0 0 512 512"><text x="50%" y="50%" dominant-baseline="middle" text-anchor="middle" font-size="20">{model_name}: {prompt}</text></svg>'
+        svg = f'<svg xmlns="http://www.w3.org/2000/svg" width="512" height="512" viewBox="0 0 512 512"><text x="50%" y="50%" dominant-baseline="middle" text-anchor="middle" font-size="20">{diffsketcher}: {prompt}</text></svg>'
         
         # Create a placeholder image
         image = Image.new('RGB', (512, 512), color = (100, 100, 100))