Trying to fix things.

app.py

@@ -21,7 +21,7 @@ tokenizer_name_gpu = "MoritzLaurer/roberta-large-zeroshot-v2.0-c"
 classifier_cpu = pipeline(task="zero-shot-classification", model=model_name_cpu, tokenizer=tokenizer_name_cpu, device="cpu", torch_dtype=torch.bfloat16)
 classifier_gpu = pipeline(task="zero-shot-classification", model=model_name_gpu, tokenizer=tokenizer_name_gpu, device="cuda", torch_dtype=torch.bfloat16) if torch.cuda.is_available() else classifier_cpu
 
-def classify(data_string, request: gradio.Request):
+def predict(data_string):
 
     data = json.loads(data_string)
 
@@ -53,8 +53,7 @@ def create_sequences(data):
     return [data['sequence'] + '\n' + data['hypothesis_template'].format(label) for label in data['candidate_labels']]
 
 gradio_interface = gradio.Interface(
-    fn = classify,
-    api_name="predict",
+    fn = predict,
     inputs = gradio.Textbox(label="JSON Input"),
     outputs = gradio.Textbox(label="JSON Output"),
     title = "Statosphere Backend",