Update app.py

app.py

@@ -4,6 +4,18 @@ from huggingface_hub import InferenceClient
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
+from transformers import pipeline
+
+model_name = "EleutherAI/gpt-j-6B"
+generator = pipeline("text-generation", model=model_name)
+
+def generate_text(prompt):
+    return generator(prompt, max_length=50)[0]["generated_text"]
+
+interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
+interface.launch()
+
+
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")