Update app.py

app.py

@@ -52,12 +52,16 @@ def predict(text):
   return pipe(text)[0]["translation_text"]
  
 def main():
+    if torch.cuda.is_available():
+        result = "True"
+    else:
+        result = "False"
     load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f")
     iface = gr.Interface(
       fn=predict, 
       inputs='text',
       outputs='text',
-      examples=[["Time-TravelRephotography"]]
+      examples=[[result]]
     )
     
     iface.launch()