Update app.py

app.py

@@ -1,29 +1,14 @@
 import gradio as gr
-from PIL import Image
-import torch
-from transformers import ViTFeatureExtractor, ViTForImageClassification
+from flask import Flask
 
-# Load model + processor
-model = ViTForImageClassification.from_pretrained("Falconsai/nsfw_image_detection")
-processor = ViTFeatureExtractor.from_pretrained("Falconsai/nsfw_image_detection")
+app = Flask(__name__)
 
-def predict(image):
-    # Process image
-    inputs = processor(images=image, return_tensors="pt")
-    with torch.no_grad():
-        outputs = model(**inputs)
-        logits = outputs.logits
-        predicted_class_idx = logits.argmax(-1).item()
-        label = model.config.id2label[predicted_class_idx]
-    return label
+def greet(name):
+    return f"Hello, {name}!"
 
-# Gradio Interface
-demo = gr.Interface(
-    fn=predict,
-    inputs=gr.Image(type="pil", label="Upload Image"),
-    outputs=gr.Label(label="Prediction"),
-    title="NSFW Image Detection (ViT)",
-    description="Upload an image to classify if it's NSFW or SFW."
-)
+gradio_interface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
-demo.launch(debug=True, server_name="0.0.0.0", server_port=7860)
+gr.mount_gradio_app(app, gradio_interface, path="/gradio")
+
+if __name__ == "__main__":
+    app.run()