Update app.py

app.py

@@ -5,26 +5,22 @@ import requests
 API_URL = "https://api-inference.huggingface.co/models/nasa-impact/nasa-smd-ibm-v0.1"
 headers = {"Authorization": "Bearer HF_TOKEN"}
 
-def query_huggingface():
-    payload = {
-        "inputs": "The answer to the universe is <mask>."
-    }
+def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
 def get_model_output():
-    response = query_huggingface()
+    payload = {
+        "inputs": "The answer to the universe is <mask>."
+    }
+    response = query(payload)
     # Extract the output from the response
     output = response.get('generated_text', 'No response')  # Adjust according to the model's response format
     return output
 
-# Gradio interface
-def gradio_interface():
-    return get_model_output()
-
-# Define Gradio blocks
+# Define Gradio interface
 demo = gr.Interface(
-    fn=gradio_interface,
+    fn=get_model_output,
     inputs=[],
     outputs="text",
     title="Hugging Face Model Output",
@@ -35,3 +31,4 @@ demo = gr.Interface(
 demo.launch()
 
 
+