Upload app.py

app.py

@@ -42,13 +42,13 @@ def instruct_generate(
         top_k: The number of top most probable tokens to consider in the sampling process.
         temperature: A value controlling the randomness of the sampling process. Higher values result in more random
     """
-    scene_name = os.path.basename(img_path).split(".")[0]
+    # scene_name = os.path.basename(img_path).split(".")[0]
     # if input in input_value_list:
     #     input = example_dict[scene_name]["input"]
     if "..." in input:
         input = input.replace("...", "")
     output = [prompt, input, max_new_tokens, top_k, temperature]
-    print(scene_name)
+    print(img_path)
     return output
 
 # 配置具体参数