siddartha1506 commited on
Commit
1414894
·
verified ·
1 Parent(s): 7e0fe33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -2,17 +2,14 @@ import gradio as gr
2
  import requests
3
  import os
4
 
5
- # Get your API key securely from Hugging Face Space Secrets
6
  API_URL = "https://api-inference.huggingface.co/models/TripoSR/TripoSR"
7
  headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
8
 
9
  def generate_3d(image):
10
  try:
11
- # Send image to TripoSR model API
12
  with open(image, "rb") as f:
13
  response = requests.post(API_URL, headers=headers, files={"file": f})
14
-
15
- # Check response
16
  if response.status_code == 200:
17
  return response.content
18
  else:
@@ -20,14 +17,12 @@ def generate_3d(image):
20
  except Exception as e:
21
  return f"An error occurred: {str(e)}"
22
 
23
- # Build Gradio interface
24
  iface = gr.Interface(
25
  fn=generate_3d,
26
  inputs=gr.Image(type="filepath", label="Upload 2D Image"),
27
  outputs="model3d",
28
  title="Scan Spectrum - 2D to 3D Converter",
29
- description="Upload any 2D image to generate a realistic 3D model that you can rotate and zoom.",
30
  )
31
 
32
- # Launch the app
33
  iface.launch()
 
2
  import requests
3
  import os
4
 
5
+ # Get Hugging Face API key from the secret you just saved
6
  API_URL = "https://api-inference.huggingface.co/models/TripoSR/TripoSR"
7
  headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
8
 
9
  def generate_3d(image):
10
  try:
 
11
  with open(image, "rb") as f:
12
  response = requests.post(API_URL, headers=headers, files={"file": f})
 
 
13
  if response.status_code == 200:
14
  return response.content
15
  else:
 
17
  except Exception as e:
18
  return f"An error occurred: {str(e)}"
19
 
 
20
  iface = gr.Interface(
21
  fn=generate_3d,
22
  inputs=gr.Image(type="filepath", label="Upload 2D Image"),
23
  outputs="model3d",
24
  title="Scan Spectrum - 2D to 3D Converter",
25
+ description="Upload any 2D image to generate a realistic 3D model you can rotate and zoom.",
26
  )
27
 
 
28
  iface.launch()