test

app.py

@@ -4,8 +4,7 @@ import torch
 import subprocess
 import numpy as np
 
-print(f"Is CUDA available: {torch.cuda.is_available()}")
-print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+
 
 
 # Function to start the ochat server
@@ -26,6 +25,8 @@ def start_ochat_server():
 # Function to interact with the chat server
 @spaces.GPU
 def chat_with_ochat(message):
+    print(f"Is CUDA available: {torch.cuda.is_available()}")
+    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
     # Here you would add the code to interact with the ochat server
     # For simplicity, this is just a placeholder response
     return "Response from ochat server"
@@ -43,4 +44,4 @@ iface = gr.Interface(
     description="Type your message and get a response from the ochat server."
 )
 
-iface.launch(share=True)
+iface.launch()