Update app.py

app.py

@@ -17,7 +17,7 @@ def print_gpu_utilization():
     nvmlInit()
     handle = nvmlDeviceGetHandleByIndex(0)
     info = nvmlDeviceGetMemoryInfo(handle)
-    print(f"GPU memory occupied: {info.used//1024**2} MB.")
+    return f"GPU memory occupied: {info.used//1024**2} MB."
 
 
 def print_summary(result):
@@ -31,12 +31,12 @@ dummy_data = {
 }
 ds = Dataset.from_dict(dummy_data)
 ds.set_format("pt")
-
+result = print_gpu_utilization()
 iface = gr.Interface(
   fn=predict, 
   inputs='text',
   outputs='text',
-  examples=[[f'result']]
+  examples=[f'{result}']
 )
 
 iface.launch()