t

app.py

@@ -20,15 +20,18 @@ import gradio as gr
 
 import requests
 def f(text):
-    context = text
-    payload = {
-        "context": context,
-        "token_max_length": 512,
-        "temperature": 1.0,
-        "top_p": 0.9,
-    }
-    response = requests.post("http://api.vicgalle.net:5000/generate", params=payload).json()
-    # http://api.vicgalle.net:5000/docs#/default/generate_generate_post
+    try:
+        context = text
+        payload = {
+            "context": context,
+            "token_max_length": 512,
+            "temperature": 1.0,
+            "top_p": 0.9,
+        }
+        response = requests.post("http://api.vicgalle.net:5000/generate", params=payload).json()
+        # http://api.vicgalle.net:5000/docs#/default/generate_generate_post
+    except Exception as e:
+        return "error: \n"+str(e)
     return response.text
 
    
@@ -39,13 +42,13 @@ iface.launch(enable_queue=True)
 # all below works but testing
 # import gradio as gr
 
-title = "GPT-J-6B"
+# title = "GPT-J-6B"
 
-examples = [
-    ['The tower is 324 metres (1,063 ft) tall,'],
-    ["The Moon's orbit around Earth has"],
-    ["The smooth Borealis basin in the Northern Hemisphere covers 40%"]
-]
+# examples = [
+#     ['The tower is 324 metres (1,063 ft) tall,'],
+#     ["The Moon's orbit around Earth has"],
+#     ["The smooth Borealis basin in the Northern Hemisphere covers 40%"]
+# ]
 
 # gr.Interface.load("huggingface/EleutherAI/gpt-j-6B",
 #     inputs=gr.inputs.Textbox(lines=10, label="Input Text"),

requirements.txt

@@ -1,4 +1 @@
-transformers
-flax
-tensorflow
-torch
+requests