Seniordev22 commited on
Commit
2bb163a
·
verified ·
1 Parent(s): 5ab0443

Update bald_processor.py

Browse files
Files changed (1) hide show
  1. bald_processor.py +1 -0
bald_processor.py CHANGED
@@ -2,6 +2,7 @@ from fastapi import FastAPI, UploadFile, File, HTTPException
2
  from fastapi.responses import StreamingResponse
3
  import io
4
  import logging
 
5
  from bald_processor import make_realistic_bald # assuming this is your custom module
6
  import gradio as gr
7
  # Logging setup (Space logs mein clear dikhega)
 
2
  from fastapi.responses import StreamingResponse
3
  import io
4
  import logging
5
+ import torch
6
  from bald_processor import make_realistic_bald # assuming this is your custom module
7
  import gradio as gr
8
  # Logging setup (Space logs mein clear dikhega)