PoseyATX commited on
Commit
fc90694
·
1 Parent(s): d70cff3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -3,6 +3,12 @@ import numpy as np
3
  import gradio as gr
4
  import requests
5
 
 
 
 
 
 
 
6
  def greet(payload):
7
  response = requests.post("https://api-inference.huggingface.co/models/PoseyATX/Fenrir_Alpha", headers={"Authorization": "Bearer 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"}, json=payload)
8
  return response.json()
 
3
  import gradio as gr
4
  import requests
5
 
6
+ import torch
7
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
8
+ # True
9
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
10
+ # Tesla T4
11
+
12
  def greet(payload):
13
  response = requests.post("https://api-inference.huggingface.co/models/PoseyATX/Fenrir_Alpha", headers={"Authorization": "Bearer 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"}, json=payload)
14
  return response.json()