Update app.py

app.py

@@ -1,16 +1,15 @@
 import gradio as gr
-import torch  # 🔥 REQUIRED even for TF pipelines
+import torch
 from transformers import pipeline
-import tensorflow as tf
-import tf_keras
 
 MODEL_ID = "Amitesh007/text_generation-finetuned-gpt2"
 
-# Load model once
+# ✅ Load PyTorch model explicitly
 generator = pipeline(
     task="text-generation",
     model=MODEL_ID,
-    framework="tf"
+    framework="pt",     # 🔑 FIX
+    device=-1           # CPU (use 0 for GPU)
 )
 
 def generate(text):
@@ -64,4 +63,6 @@ with gr.Blocks(title="GPT-2 Text Generator") as demo:
     )
 
 demo.queue(concurrency_count=2)
+demo.launch()
+