Update app.py

app.py

@@ -1,15 +1,18 @@
+
+
 import gradio as gr
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+
+
+model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased")
+tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
 
-tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
-model = GPT2LMHeadModel.from_pretrained("gpt2")
 
-def generate_text(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt").to('cuda')
-    outputs = model.generate(inputs, do_sample=False, max_length=100)
+def generate_text(input_text):
+    input_ids = tokenizer.encode(input_text, return_tensors="pt")
+    output = model.generate(input_ids)
+    return tokenizer.decode(output[0], skip_special_tokens=True)
 
-    generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return generated_text
+interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
 
-iface = gr.Interface(fn=generate_text, inputs=[gr.Textbox(label="Enter a prompt")])
-iface.launch()
+interface.launch()