Update app.py

app.py

@@ -45,7 +45,7 @@ def upscale_chunk(chunk, model, processor, device):
     return Image.fromarray(output_image)
 
 @spaces.GPU
-def main(image, model_choice, save_as_jpg=True, use_tiling=True):
+def main(image, original_filename, model_choice, save_as_jpg=True, use_tiling=True):
     # Resize the input image
     image = resize_image(image)
     
@@ -79,8 +79,8 @@ def main(image, model_choice, save_as_jpg=True, use_tiling=True):
         upscaled_image = upscale_chunk(image, model, processor, device)
     
     # Generate output filename
-    original_filename = os.path.splitext(image.filename)[0] if image.filename else "image"
-    output_filename = f"{original_filename}_upscaled"
+    original_basename = os.path.splitext(original_filename)[0] if original_filename else "image"
+    output_filename = f"{original_basename}_upscaled"
     
     if save_as_jpg:
         output_filename += ".jpg"
@@ -93,7 +93,8 @@ def main(image, model_choice, save_as_jpg=True, use_tiling=True):
 
 def gradio_interface(image, model_choice, save_as_jpg, use_tiling):
     try:
-        result = main(image, model_choice, save_as_jpg, use_tiling)
+        original_filename = getattr(image, 'name', 'image')
+        result = main(image, original_filename, model_choice, save_as_jpg, use_tiling)
         return result, None
     except Exception as e:
         return None, str(e)