Update app.py

app.py

@@ -1,17 +1,13 @@
-from gradio_client import Client
+from transformers import GPT2Tokenizer, GPT2Model
+tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
+model = GPT2Model.from_pretrained('gpt2')
+import gradio as gr
 
-# Initialize the client and set the API endpoint URL
-client = Client("https://stabilityai-stablelm-tuned-alpha-chat.hf.space/")
-client.predict(
-				fn_index=5
-)
-file1 = client.predict(fn_index=6)
-# Send a "start" message to begin the conversation
-result = client.predict("start", file1, fn_index=0)
-conversation_id = result[1]
+def gpt2(string):
+    text = string
+    encoded_input = tokenizer(text, return_tensors='pt')
+    output = model(**encoded_input)
+    return output
 
-# Send a chat message and get the response
-while True:
-    message = input("You: ")
-    result = client.predict(message, conversation_id, fn_index=2)
-    print("Bot:", open(result[1],'r').read())
+iface = gr.Interface(fn=gpt2, inputs="text", outputs="text")
+iface.launch()