Update Dockerfile

Dockerfile

@@ -1,36 +1,23 @@
-import gradio as gr
-import subprocess
+# Use the Hugging Face Docker image as a parent image
+FROM huggingface/autotrain-advanced:latest
 
-def deblur_image(input_image_path, angle_model_path, length_model_path):
-    command = [
-        "python",
-        "deblur_img.py",
-        "--image",
-        input_image_path,
-        "--angle_model",
-        angle_model_path,
-        "--length_model",
-        length_model_path
-    ]
-    result = subprocess.run(command, capture_output=True, text=True)
-    return result.stdout
+# Install Gradio
+RUN pip install gradio
 
-def deblur_and_display(input_image, angle_model_path, length_model_path):
-    output = deblur_image(input_image.name, angle_model_path, length_model_path)
-    return output
+# Set the working directory
+WORKDIR /app
 
-iface = gr.Interface(
-    fn=deblur_and_display,
-    inputs=[
-        gr.inputs.Image(label="Upload Image"),
-        gr.inputs.Textbox(label="Angle Model Path", default="/app/pretrained_models/angle_model.hdf5"),
-        gr.inputs.Textbox(label="Length Model Path", default="/app/pretrained_models/length_model.hdf5")
-    ],
-    outputs="text",
-    title="Image Deblurring App",
-    description="Upload an image to deblur it."
-)
+# Install git
+RUN apt-get update && apt-get install -y git
 
-if __name__ == "__main__":
-    iface.launch()
-    
+# Clone the repository
+RUN git clone https://github.com/ankitVP77/Blind-Motion-Deblurring-for-Legible-License-Plates-using-Deep-Learning.git /app
+
+# Install dependencies
+RUN pip install --no-cache-dir tensorflow numpy opencv-python-headless matplotlib
+
+# Expose the port the app runs on
+EXPOSE 7860
+
+# Command to launch Gradio interface
+CMD ["python", "app.py"]