Update app.py

Trying to get it to work for CPU mode

app.py

@@ -99,12 +99,17 @@ def evaluate(
             yield out_str.strip()
             out_last = i + 1
 
-    gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
-    print(f'vram {gpu_info.total} used {gpu_info.used} free {gpu_info.free}')  
+    if HAS_GPU == True :
+        gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
+        print(f'vram {gpu_info.total} used {gpu_info.used} free {gpu_info.free}')  
+    
     del out
     del state
     gc.collect()
-    torch.cuda.empty_cache()
+
+    if HAS_GPU == True :
+        torch.cuda.empty_cache()
+    
     yield out_str.strip()
 
 # Examples and gradio blocks